[codemog]

Curious why the model architecture wasn’t talked about at all? Did I miss that part?

[g58892881]

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