卷积神经网络中的训练和测试

时间:2020-05-23 14:35:13

标签: tensorflow keras conv-neural-network

我想知道在什么阶段使用CNN?是在训练期间每批或一个纪元完成后使用,还是在所有纪元完成后使用?我对这两个过程如何一起运行感到有些困惑?类似地,在每批或每个时期之后进行梯度更新?

model.fit_generator(
aug.flow(x_train, y_train, batch_size=BATCH_SIZE),
validation_data=(x_test, y_test),
steps_per_epoch=len(x_train) // BATCH_SIZE,
epochs=EPOCHS, verbose=1, callbacks = callbacks)

从fit_generator中可以清楚地看到,图像是逐批加载到内存中的。

1 个答案:

答案 0 :(得分:1)

Keras在每个时期的末尾使用验证数据集(如果您没有在validation_freq函数中更改fit的话)。模型在整个训练数据集上进行训练的每个时期,然后在验证数据集上进行评估