ipython下运行代码出现<module 'main' (built-in)> is a built-in class错误

tang-0203

于 2017-11-22 21:53:51 发布

阅读量4.8k

点赞数 1

分类专栏： Pytorch学习模型剪枝文章标签： python pytorch 加载模型出错查看模型参数是否训练模型剪枝

本文链接：https://blog.youkuaiyun.com/tsq292978891/article/details/78608735

版权

在进行模型剪枝时，作者在ipython环境下载入VGG16并修改全连接层以适应数据，训练后保存模型。但在之后加载模型时遇到内置类错误。问题可能源于Python的一个Bug。社区讨论指出，`torch.save`保存了包括网络结构、参数和训练状态等所有信息。解决方案是在加载模型前检查各层的requires_grad属性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个问题是我在做模型剪枝的时候遇到的

先载入VGG16模型，更改全连接层，使输出单元个数与自己待处理的数据类别个数一致。
冻住卷积层，只训练全连接层。学习率设为0.0001，momentum=0.9。
当设定的epoch数目满足的时候，把模型保存，使用命令：

torch.save(model, "./cifar-10/5epochs_cifar10_vgg_model")

4 . 然后在ipython使用以下命令加载模型：

model = torch.load("/home/smiles/tsq/PyTorch/pytorch-pruning/cifar-10/5epochs_cifar10_vgg_model").cuda()

这个时候就出现错误：

<module '__main__' (built-in)> is a built-in class

关于这个问题：网上有很多人遇到。应该是一个Bug。
python社区有关于这个问题的讨论：

最低0.47元/天解锁文章

ipython下运行代码出现<module '__main__' (built-in)> is a built-in class错误

ipython下运行代码出现<module 'main' (built-in)> is a built-in class错误