我是靠谱客的博主 寂寞花瓣,最近开发中收集的这篇文章主要介绍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所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(35)

评论列表共有 0 条评论

立即
投稿
返回
顶部