File size: 769 Bytes
a3ed971 7160924 1ade0f2 7160924 56a6fc9 7160924 a3ed971 7160924 56a6fc9 7160924 56a6fc9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
---
framework: pytorch
model: EfficientNet-B1
dataset: CIFAR10 (restructured to have random upright and upside-down samples, with labels {0:'up', 1:'down'}
imgSize: 224x224x3
---
This repository contains model trained to predict orientation {0:'up', 1:'down'} of images.
The model '.pkl' file contains a dictionary with the following keys:
```python:
'optimizer': optimizer state dictionary
'scheduler': scheduler state dictionary
'model': model state dictionary
'epoch': checkpoint epoch
```
The image size is 3x224x224. The model can be initialized as:
```python:
model = torchvision.models.efficientnet_b1(pretrained=True)
in_features = model.classifier[1].in_features
out_features = 2
model.classifier[1] = nn.Linear(in_features, out_features, bias=True)
``` |