model.py 文件源码

python
阅读 24 收藏 0 点赞 0 评论 0

项目:pytorch-model-zoo 作者: sdhnshu 项目源码 文件源码
def forward(self, x):
        out = F.leaky_relu(self.conv1(x), 0.05)    # (?, 64, 32, 32)
        out = F.leaky_relu(self.conv2(out), 0.05)  # (?, 128, 16, 16)
        out = F.leaky_relu(self.conv3(out), 0.05)  # (?, 256, 8, 8)
        out = F.leaky_relu(self.conv4(out), 0.05)  # (?, 512, 4, 4)
        out = F.sigmoid(self.conv5(out)).squeeze()
        return out
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号