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'