如何解决在pytorch中实现DCGAN时出错:内核大小不能大于实际输入大小
我一直在尝试在 PyTorch 中训练 DCGAN 实现。我收到以下错误:
运行时错误:计算出的每个通道的填充输入大小:(3 x 3)。内核大小:(4 x 4)。内核大小不能大于实际输入大小
我修改了 this implementation 中定义的鉴别器网络。
修改后的判别器网络如下:
self.main = nn.Sequential(
nn.Conv2d(nc,ndf,4,2,1,bias=False),nn.LeakyReLU(0.2,inplace=True),nn.Conv2d(ndf,ndf * 2,nn.Batchnorm2d(ndf * 2),nn.Conv2d(ndf * 2,ndf * 4,nn.Batchnorm2d(ndf * 4),nn.Conv2d(ndf * 4,ndf * 8,nn.Batchnorm2d(ndf * 8),nn.Conv2d(ndf * 8,ndf * 16,nn.Batchnorm2d(ndf * 16),nn.Conv2d(ndf * 16,nn.Sigmoid()
)
提前致谢!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。