inarikami / keras-rl2

Reinforcement learning with tensorflow 2 keras
MIT License
251 stars 105 forks source link

Problem logging with tensorboard #29

Open Carterbouley opened 3 years ago

Carterbouley commented 3 years ago

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!

MarcoLehmann commented 3 years ago

same here

j-beaver commented 3 years ago

same here