pytorch Bug Reaper

Posted on 2018-01-Mon Edited on 2018-04-Sun Views: Views:

net结构文件找不到

在main.py 中import

load失败

torch.save(the_model.state_dict(), PATH)

the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

load UnicodeDecodeError

from functools import partial
import pickle
pickle.load = partial(pickle.load, encoding="utf8")
pickle.Unpickler = partial(pickle.Unpickler, encoding="utf8")
model = torch.load(model_file, map_location=lambda storage, loc: storage, pickle_module=pickle)

https://github.com/CSAILVision/places365/issues/25