我是靠谱客的博主 寂寞花瓣,最近开发中收集的这篇文章主要介绍Bug解决-Torch.save FileNotFoundError: [Errno 2] No such file or directory,觉得挺不错的,现在分享给大家,希望可以做个参考。
概述
Bug:
model_state_dict = {
key.replace('module.', ''): tmp[key] for key in tmp if
(('SearchNet' not in key) and ('_copy' not in key))
}
model_name = self.args.save_dir.strip('/')+'/model/model_latest.pth'
torch.save(model_state_dict, model_name)
Torch.save FileNotFoundError: [Errno 2] No such file or directory
解决方法:
原先代码中default设置的路径是绝对路径,现在改为相对路径没有报错
parser.add_argument('--save_dir', type=str, default='./log/',
help='Directory to save log, arguments, models and images')
最后
以上就是寂寞花瓣为你收集整理的Bug解决-Torch.save FileNotFoundError: [Errno 2] No such file or directory的全部内容,希望文章能够帮你解决Bug解决-Torch.save FileNotFoundError: [Errno 2] No such file or directory所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复