最近用了一下paddlepaddle深度学习框架训练了一下目标检测模型,然后在训练一段时候后在验证的时候出现了错误: File "...\PaddleDetection-master\ppdet\utils\coco_eval.py", line 318, in bbox2out catid = (clsid2catid[int(clsid)]) KeyError: 6 小编是训练一个目标检测的模型, KeyError: 6 中的6是要检测的类别的数量(6个类别)。 遇到这个错误的时候,很多人可能要放弃了(…

方法一: import paddle.fluid print(paddle.fluid.install_check.run_check()) 方法二(后测试有时没用): import paddle.fluid place = paddle.fluid.CUDAPlace(0) print(place)

保存模型 model_save_path="model/mnist-model/dygraph-mnist" fluid.save_dygraph(model.state_dict(),model_save_path)#保存模型参数 fluid.save_dygraph(optimaizer.state_dict(),model_save_path)#保存优化器参数 加载模型 params_dict, opt_dict = fluid.load_dygraph(model_save_path) model = MN…

paddle.fluid.dygraph.save_dygraph(state_dict, model_path) 该接口将传入的参数或优化器的 dict 保存到磁盘上。 state_dict 是通过 Layer 的 state_dict() 方法得到的。 注: model_path 不可以是一个目录。 该接口会根据 state_dict 的内容,自动给 model_path …