查看模型每层及总信息

最新推荐文章于 2023-05-04 16:54:29 发布

周小天..

最新推荐文章于 2023-05-04 16:54:29 发布

阅读量263

点赞数 1

分类专栏：图像处理

本文链接：https://blog.csdn.net/xingghaoyuxitong/article/details/117674067

版权

图像处理专栏收录该内容

53 篇文章 3 订阅

订阅专栏

import torch
from torch.nn import Module
from torchsummary import summary
class lrelu_conv_bn(Module):
    def __init__(self):
        super(lrelu_conv_bn,self).__init__()
        self.conv = torch.nn.ConvTranspose2d(1,1,kernel_size=4,stride=2,padding=1,bias=False)

    def forward(self,x):
        x = self.conv(x)
        return x

model = lrelu_conv_bn()
summary(model=model,input_size = (1,2,2),batch_size=1,device='cpu')

输出：