ssa*_*ahu 2 deep-learning keras tensorflow
我正在尝试训练VGG16模型并拥有4k训练样本和2k验证样本.
在上图中,即使最初的134批次完成得非常快,最后一批等待很长时间并在大约8分钟后完成,我认为这花费的时间太长.我有什么不对劲吗?使用以下内容开始培训过程.
model.compile(optimizer='rmsprop', loss='categorical_crossentropy',metrics=['accuracy'])
model.fit_generator(train_generator, samples_per_epoch=4320, epochs=50, validation_data=validation_generator, nb_val_samples=2880)
| 归档时间: |
|
| 查看次数: |
2367 次 |
| 最近记录: |