后台运行不能保存模型的问题
收藏
paddle.save(model.state_dict(), '/home/aistudio/work/bestmodel/yolo_best{}.pdparams'.format(epoch))
我在项目中是用这个保存代码,后台运行的输出结果是对的,但是,运行文件里面却没有应该保存的文件,这是bug吗?在项目里面运行是可以的
0
收藏
请登录后评论
你看看位置,文档里好像是规定了输出位置
可是,我是用notebook不是用脚本呀?也需要特定事物输出位置吗?
提个issue问问吧