【发布时间】:2018-05-19 19:25:02
【问题描述】:
我已经检查了所有解决方案,但我仍然面临同样的错误。我的训练图像形状是(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)
【问题讨论】:
标签: python tensorflow deep-learning keras conv-neural-network