【发布时间】:2019-06-12 15:22:02
【问题描述】:
我在运行facebookresearch的maskrcnn-benchmark官方代码时遇到的问题,加载预训练模型时出错。
代码在学校的远程服务器上运行,显卡是 NVIDIA P100。
检查指针 = DetectronCheckpointer( cfg、模型、优化器、调度器、output_dir、save_to_disk) extra_checkpoint_data = checkpointer.load(cfg.MODEL.WEIGHT) arguments.update(extra_checkpoint_data)
我希望正确运行代码并理解为什么会出现此问题。
【问题讨论】:
标签: python-3.x pytorch