Curious why the model architecture wasn’t talked about at all? Did I miss that part?
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#...
alt Hacker News
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#...