Keras fit生成器不在Tensorflow后端使用GPU

时间:2017-10-24 15:13:22

标签: tensorflow keras

我有一个用Keras实现的代码,我正在使用fit_generator,因为数据的大小非常大。

这是我的代码:

model = ad.kerasCNN() 

model.fit_generator(
    ad.data_generator(X_TRAIN_metadata, Y_TRAIN_metadata, batch_size), 
    steps_per_epoch, 
    epochs=maxEpoch, 
    verbose=1, 
    callbacks=[history, early_stopping], 
    validation_data=ad.data_generator(X_DEV_metadata, Y_DEV_metadata, batch_size), 
    validation_steps=steps_per_epoch_dev, 
    class_weight=None, 
    max_queue_size=10, 
    workers=1, 
    use_multiprocessing=False, 
    initial_epoch=0
)

当我使用 GPU 运行此代码时,我在训练阶段没有遇到任何问题,GPU的使用率在70-80%之间。

但是当代码进入验证时,不使用GPU。 这意味着验证时间比训练阶段长23倍。

这是keras / tensorflow的正常行为吗?

在验证时间内如何使用GPU?

0 个答案:

没有答案