如何修复ValueError:输入通道的数量与筛选器的相应维度不匹配,3!=64



我几乎100%确信这是让我现在的生活变得艰难的代码:

model.add(Convolution2D(128, 5, 5, subsample=(2, 2), input_shape=(3, 64, 64), border_mode = 'same'))

很抱歉简短的描述,但我不知道在这里该怎么办,我对GANs和机器学习很陌生。

@Amit Vikram Singh的答案是有效的,我想进一步补充一点,你应该把它放在try/except代码块中,这样你的程序在遇到类似错误时就不会崩溃。你可以使用这样的东西:

try:
model.add(Convolution2D(128, 5, 5, subsample=(2, 2), input_shape=(64, 64, 3), border_mode = 'same'))
except ValueError:
# code to run if it fails here

最新更新