3 ms·
Curious why the model architecture wasn’t talked about at all? Did I miss that part?
by codemog 7mo ago
Curious why the model architecture wasn’t talked about at all? Did I miss that part?
- g58892881 7mo agopretty 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#...