Keras - on_batch_end 方法很慢，但我只有回调是检查点

Question

我使用 TensorFlow 后端与 keras 建立了一个网络。

当我训练我的网络时，我经常不断收到消息：

UserWarning: Method on_batch_end() is slow compared to the batch update (0.195523). Check your callbacks.
  % delta_t_median)

问题是我的网络只设置了检查点回调：

checkpoint = ModelCheckpoint(filepath, monitor='val_loss', verbose=1, save_best_only=True, mode='min')
callbacks_list = [checkpoint]

据我在文档中看到的这个方法只在纪元结束时被调用，所以它不能减慢on_batch_end方法的速度。任何人都可以提供一些有关问题的信息吗？

Answer 1

这很可能是 Generator ( fit_generator()) 问题。使用生成器作为数据源时，必须在批处理结束时调用它。考虑重新访问您的生成器代码，使用多处理 ( workers > 1) 或更高的批处理大小（如果可能）