大佬教程收集整理的这篇文章主要介绍了pytorch 或其他机器学习库中是否有 model.vgg16 命名模块?,大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。
我认为pytorch提供的vgg16模块是@H_802_3@models.vgg16,包括's'。但是,当我尝试使用我自己实现的 vgg16 模型进行训练时,似乎与其他 vgg16 模型或其他模型存在一些冲突。
...
from model.vgg16 import VGG16
...
这就是我在 Train.py
中导入自定义 vgg16 模型的方式。 vgg16.py
文件位于 @H_802_3@model 文件夹中,该文件夹与 Train.py
位于同一目录中。如果我使用 %run Train.py
命令运行它,它会输出 maxpool2d 错误(输出太小),即使我删除了 vgg16 模型的所有内容(等、变量、函数),只保留类实现。
然后,当我将自定义 vgg16 从 vgg16.py
重命名为 vgg15.py
或其他(甚至是像 vgg19 这样的现有名称)时,将导入实现更改为:
...
from model.vgg15 import VGG16
...
好吧,现在运行没有任何错误。似乎只有 @H_802_3@model.vgg16 发生错误。我不明白为什么会这样。在 @H_802_3@model 文件夹中没有任何其他文件夹或文件。
此外,当我尝试使用 Train.py
而不是 !python
运行 %run
时,虽然从 @H_802_3@model.vgg16 导入也没有错误,但这无法打印测试图像或学习图表损失。
当我尝试使用检查点测试我的模块时发生完全相同的情况,现在输出数据集大小不匹配错误...
为什么会这样?
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)
以上是大佬教程为你收集整理的pytorch 或其他机器学习库中是否有 model.vgg16 命名模块?全部内容,希望文章能够帮你解决pytorch 或其他机器学习库中是否有 model.vgg16 命名模块?所遇到的程序开发问题。
如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。