python - keras 张量板回调，没有属性“_should_trace”

Question

我正在尝试使用 keras-rl2 库来安装 DQN。我相信回调应该与训练任何神经网络一样工作。

我想可视化代理的训练，当我调用 fit 方法打开 tensorbaord 来查看它的进度。

from keras.callbacks import TensorBoard
tb = TensorBoard(log_dir='/keras-rl')
dqn.fit(env, nb_steps=1200000, visualize=False, verbose=1, callbacks=[tb])


AttributeError: 'TensorBoard' object has no attribute '_should_trace'

我搜索了 _should_trace 错误，但找不到任何对它的引用。有人知道解决方法吗，或者我是否应该以不同的方式构造我的回调？

标签： pythontensorflowkerastensorboard