4 ms·
pretty basic ``` from torchvision import models # Avoid downloading pretrained weights; we load trained checkpoint weights. model = models.resnet18(weights=No
by g58892881 6mo ago
pretty basic
```
from torchvision import models
# Avoid downloading pretrained weights; we load trained checkpoint weights.
model = models.resnet18(weights=None)
model.fc = nn.Linear(model.fc.in_features, num_classes)
```
https://github.com/mixfont/lens/blob/main/lens_inference.py#L293 https://github.com/mixfont/lens/blob/main/lens_inference.py#...