keras-rl2 icon indicating copy to clipboard operation
keras-rl2 copied to clipboard

Problem logging with tensorboard

Open Carterbouley opened this issue 4 years ago • 2 comments

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!

Carterbouley avatar Feb 19 '21 11:02 Carterbouley

same here

MarcoLehmann avatar Mar 09 '21 09:03 MarcoLehmann

same here

j-beaver avatar May 09 '21 21:05 j-beaver