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)