logoalt Hacker News

g58892881yesterday at 8:54 PM0 repliesview on HN

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