运行cycleGAN的test.py时遇到RuntimeError: Error(s) in loading state_dict for ResnetGenerator

解决PyTorch中ResnetGenerator加载错误：去除module.前缀

最新推荐文章于 2024-09-30 11:03:09 发布

原创

最新推荐文章于 2024-09-30 11:03:09 发布 · 439 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #生成对抗网络 #pycharm #经验分享

运行cycleGAN的test.py时遇到报错RuntimeError: Error(s) in loading state_dict for ResnetGenerator

发生这个错误的时候，是加载networks的时候出现问题，我的情况是发现实际上加载keys的时候多了module.，故只需要在加载的时候把它去掉就好。

找到工程里的models/base_model.py文件，找到196行左右的def load_networks(self, epoch)：

原代码为：

    def load_networks(self, epoch):
        """Load all the networks from the disk.

        Parameters:
            epoch (int) -- current epoch; used in the file name '%s_net_%s.pth' % (epoch, name)
        """
        for name in self.model_names:
            if isinstance(name, str):
                load_filename = '%s_net_%s.pth' % (epoch, name)
                if self.opt.isTrain and self.opt.pretrained_name is not None:
                    load_dir = os.path.join(self.opt.checkpoints_dir, self