小编yan*_*guo的帖子

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

我在运行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)

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

python-3.x pytorch

0
推荐指数
1
解决办法
1万
查看次数

标签 统计

python-3.x ×1

pytorch ×1