a-whitej

Results 11 comments of a-whitej

For pytorch issue(on Python 3.5.2) Add: from functools import partial import pickle pickle.load = partial(pickle.load, encoding="latin1") pickle.Unpickler = partial(pickle.Unpickler, encoding="latin1") And make change: # load pretrained model load_model(torch.load('./models/alexnet.pth',pickle_module=pickle), net)