UnpicklingError: неверный ключ загрузки, '\ x0a' - PullRequest
Новая
       26

UnpicklingError: неверный ключ загрузки, '\ x0a'

0 голосов
/ 09 января 2019 
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'.

1 Ответ

0 голосов
/ 09 января 2019

Сохранена ли модель с использованием той же версии Python, что и та, которая пытается загрузить. Если нет, то были некоторые изменения в рассоле в разных версиях. Также, если это не так, вы можете также опубликовать вызов torch.save (), который вы делаете.

...