我已经检查了所有解决方案,但仍然面临同样的错误。我的训练图像形状是(26721, 32, 32, 1)
,我相信它是 4 维,但我不知道为什么错误显示它是 5 维。
model = Sequential()
model.add(Convolution2D(16, 5, 5, border_mode='same', input_shape= input_shape ))
这就是我的定义model.fit_generator
model.fit_generator(train_dataset, train_labels, nb_epoch=epochs, verbose=1,validation_data=(valid_dataset, valid_labels), nb_val_samples=valid_dataset.shape[0],callbacks=model_callbacks)