Open zihaozhang9 opened 4 years ago
net.load_state_dict(torch.load('./model/net.pth'))
you can get it by scripts/get_state_dict.py.
FileNotFoundError: [Errno 2] No such file or directory: './model/resnet50.pth'
net.load_state_dict(torch.load('./model/net.pth'))