keras-rl2
keras-rl2 copied to clipboard
Problem logging with tensorboard
I am trying to log the training of my agent over time.
I am used to using tensorboard, however when I try and create a callback, i get an error when running a fit.
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'
And when I try and use WandbLogger as suggested:
from keras.callbacks import WandbLogger
ImportError: cannot import name 'WandbLogger' from 'keras.callbacks'
If there is a solution to this I would be thankful, or another way of doing live monitoring would be great too!
same here
same here