https://github.com/rishikanthc/torch-gists
My collection of utilities for pytorch
https://github.com/rishikanthc/torch-gists
Last synced: 9 days ago
JSON representation
My collection of utilities for pytorch
- Host: GitHub
- URL: https://github.com/rishikanthc/torch-gists
- Owner: rishikanthc
- License: mit
- Created: 2021-05-11T23:37:57.000Z (about 5 years ago)
- Default Branch: main
- Last Pushed: 2021-05-19T06:05:14.000Z (about 5 years ago)
- Last Synced: 2025-11-28T06:44:56.501Z (7 months ago)
- Language: Python
- Size: 10.7 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
## Torch-gists
A collection of models and utilities that I frequently use in my ML projects
#### Importing models
Models included:
- ResNet (18, 34, 101, 152)
- VGG
- MobileNet-V2
To import models:
```python
from torch_gists.models import ResNet18
model = ResNet18(num_classes=10)
```
#### Utilities
##### wrapper for pytorch_lightning
A base class `pl_wrapper` that extens pytorch_lightning implementing training, validation, testing loops,
dataloader that's common to most models.
Users can extend the `pl_wrapper` class and implement their own `__ini__`, `forward` and data methods.
```python
from torch_gists.utils import pl_wrapper
class my_model(pl_wrapper):
def __init__(self):
super().__init__()
...
def forward(self, x):
...
```