print(net)打印出来的只有网络名,没有网络结构

本文档介绍了如何修复一个PyTorch实现的VGGNet模型,在尝试打印网络结构时仅显示名称的问题。问题在于`_init_`函数的书写不规范,通过将其改为`__init__`并修复初始化过程,成功实现了网络结构的打印。
摘要由CSDN通过智能技术生成

@[TOC](print(net) only prints CNN’s name not structure)

问题

按照打印网络的语句,虽然不报错,却没有打印出来网络结构,经过一番排查之后,发现了问题出在了初始化。

解决办法

_init_替换成__init__前后两个下划线,写法不规范导致的错误。

class Vggnet(nn.Module):
    #def _init_(self):
    def _init_(self):####_init_替换成__init__前后两个下划线
        super(Vggnet, self)._init_()  ####_init_替换成__init__前后两个下划线
        self.conv = nn.Sequential(nn.Conv2d(3,64,3,padding = 1),
                                 nn.ReLU(),
                                  nn.Conv2d(64,64,3),
                                  nn.MaxPool2d(2, 2),#
                                  nn.Conv2d(64,128,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(128,128,3,padding = 1),
                                  nn.ReLU(),
                                  nn.MaxPool2d(2, 2),#
                                  nn.Conv2d(128,256,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(256,256,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(256,256,3,padding = 1),
                                  nn.ReLU(),
                                  nn.MaxPool2d(2, 2),##
                                  nn.Conv2d(256,512,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(512,512,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(512,512,3,padding = 1),
                                  nn.ReLU(),
                                  nn.MaxPool2d(2, 2),##
                                  nn.Conv2d(512,512,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(512,512,3,padding = 1),
                                  nn.ReLU(),
                                  nn.Conv2d(512,512,3,padding = 1),
                                  nn.ReLU(),
                                  nn.MaxPool2d(2, 2))
        
        
        
        self.fc = nn.Sequential(nn.Linear(7*7*512,4096),
                                nn.ReLU(),
                                nn.Dropout(),
                                nn.Linear(4096,4096),
                                nn.ReLU(),
                                nn.Dropout(),
                                nn.Linear(4096,5))
                               
 
    def forward(self, x):
        x = self.conv(x)
        x = x.view(x.size(0), -1)
        x = self.fc(x)
        #output = F.softmax(x, dim=1)
        return x

net1= Vggnet()
print(net1)
==========================================

错误结果展示:
Vggnet()##只显示网络名称,不显示具体结构。
  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值