the resnet checkpoint can't be loaded

[JavanehBahrami]

Hi.
I downloaded the resnet50-epoch-15-valid_acc=-1-test_acc=0.9856.pt checkpoint but in inference part, I can not load it. It gives me an Error while loading:

device = "cuda:0"
m = torch.load('./weights/resnet50-epoch-8-valid_acc=0.97-test_acc=-1.00.pt')
m = m.to(device)
m.eval()

the error is:
in _load result = unpickler.load()
ModuleNotFoundError: No module named 'model'

[tomasz-lewicki]

Hi! This should be fixed now.
Check out the new quickstart in README 🙂