Как исправить '_pickle.UnpicklingError: неверный ключ загрузки,' <'' ошибка в Pytorch

Question

Проблема, с которой я столкнулся, когда запустил официальный код maskrcnn-benchmark для facebookresearch, что было неверно при загрузке модели предварительного обучения.

Код работает на удаленном сервере в школе и графикекарта NVIDIA P100.

checkpointer = DetectronCheckpointer (cfg, модель, оптимизатор, планировщик, output_dir, save_to_disk) extra_checkpoint_data = checkpointer.load (cfg.MODEL.WEIGHT) arguments.update (extra_checkpoint_data) arguments.update (extra_checkpoint_data)1006 * Я ожидаю, что код будет запущен правильно, и я пойму, почему это проблема.

Answer 1

Причина проблемы в том, что предыдущая загрузка не была завершена. Поэтому, когда я удалил исходный файл и повторно загрузил его, проблема была решена.