如何修复 Pytorch 中的 '_pickle.UnpicklingError: invalid load key, '<' ' 错误

Question

我在运行facebookresearch的maskrcnn-benchmark官方代码时遇到的问题，加载预训练模型时出错。

该代码在学校的远程服务器上运行，显卡是 NVIDIA P100。

checkpointer = DetectronCheckpointer（cfg，模型，优化器，调度程序，output_dir，save_to_disk）extra_checkpoint_data = checkpointer.load（cfg.MODEL.WEIGHT）arguments.update（extra_checkpoint_data）

我希望能够正确运行代码并理解为什么会出现这个问题。

Answer 1

出现这个问题的原因是之前的下载没有完成。所以当我删除了原来的文件并重新下载后，问题就解决了。