thu-ml / tianshou

An elegant PyTorch deep reinforcement learning library.
https://tianshou.org
MIT License
7.9k stars 1.12k forks source link

Wandb logger #1222

Open dev0Guy opened 1 week ago

dev0Guy commented 1 week ago

Versions:

I've the following setup of logger (terminal already connected to wandb acount & project):

wandb.init()
logger = WandbLogger()
writer = SummaryWriter('./tensorboard')
logger.load(writer)

When looking on run page inside wandb I can see only system metrics. However when run tensorboard on the SummaryWriter log path I'm able to review the result using tensorboard. I've followed logging instruction inside docs.

Thanks in advance :)


dev0Guy commented 1 week ago

Just for classification I do set the logger inside policy