UnpicklingError: 无效的加载键，'\x0a'

Question

model_save_name = 'classifier.pt'
path = F"/content/gdrive/My Drive/Others/{model_save_name}"

model.load_state_dict(torch.load(path), strict = False)

试图从路径加载到模型 state_dict。稍后将使用相同的模型 state_dict 来保存检查点，例如 torch.save(model.state_dict(), path)

错误回溯：

UnpicklingError                           Traceback (most recent call last)
<ipython-input-69-7cb7940c24e3> in <module>()
----> 1 model.load_state_dict(torch.load(path), strict = False)

/usr/local/lib/python3.6/dist-packages/torch/serialization.py in load(f, map_location, pickle_module)
    356         f = open(f, 'rb')
    357     try:
--> 358         return _load(f, map_location, pickle_module)
    359     finally:
    360         if new_fd:

/usr/local/lib/python3.6/dist-packages/torch/serialization.py in _load(f, map_location, pickle_module)
    530             f.seek(0)
    531 
--> 532     magic_number = pickle_module.load(f)
    533     if magic_number != MAGIC_NUMBER:
    534         raise RuntimeError("Invalid magic number; corrupt file?")

UnpicklingError: invalid load key, '\x0a'.

Answer 1

模型是否使用与尝试加载的模型相同的 python 版本保存。如果没有，不同版本的泡菜会发生一些变化。此外，如果不是这种情况，您还可以发布您正在进行的 torch.save() 调用。