``` 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#...