训练结束后模型权重的保存

zhenghao977 / FCOS-PyTorch-37.2AP

A pure torch implement of FCOS 37.2AP

162 stars 39 forks source link

训练结束后模型权重的保存 #9

Closed Linamiao-1998 closed 3 years ago

Linamiao-1998 commented 3 years ago

您好，博主，我的训练结束后模型权重文件并没有保存到指定文件夹中，这时如何修改呀。 torch.save(model.statedict(),"pytorchDebug/FCOS-PyTorch-37.2AP/output/model{}.pth".format(epoch + 1)) 这句话感觉没有问题呀

zhenghao977 commented 3 years ago

检查一下这个文件夹是否存在吧