问题描述
我一直在尝试在 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 (将#修改为@)