我有一个用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?