使用torchsummary报错

发布时间：2024年01月20日

源代码

from torchsummary import summary
from vggnet import VGGNet

myNet = VGGNet()
summary(myNet, (3, 28, 28))
# print(myNet)

运行报错信息

重点

说明我们应该把模型放到cuda上面去运行

修改

主要修改位置

myNet = VGGNet() =>
myNet = VGGNet().cuda()

整体

from torchsummary import summary
from vggnet import VGGNet

myNet = VGGNet().cuda()
summary(myNet, (3, 28, 28))
# print(myNet)

运行成功，解决！

特别注意点！！！

summary(myNet, (3, 28, 28))这里的(3, 28, 28)里面的3是深度，一定要和定义的一样

比如：

这里我们输入的深度是1

所以应该是summary(myNet, (1, 28, 28))

文章来源:https://blog.csdn.net/zhuyuyaya/article/details/135587837
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
最新文章