torch.save(best_model_wts,'/root/autodl-tmp/best_model.pth')
'/root/autodl-tmp/best_model.pth'是服务器中的路径,训练时根据自己的目录进行调整,不能写本地的保存路径。
'/root/autodl-tmp/best_model.pth'这条路径可以通过Xftp连接服务器后找到。
torch.save(best_model_wts,'/root/autodl-tmp/best_model.pth')
'/root/autodl-tmp/best_model.pth'是服务器中的路径,训练时根据自己的目录进行调整,不能写本地的保存路径。
'/root/autodl-tmp/best_model.pth'这条路径可以通过Xftp连接服务器后找到。
685
7008
2万+

被折叠的 条评论
为什么被折叠?