thu-ml / tianshou

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

Wandb logger #1222

Open dev0Guy opened 1 month ago

dev0Guy commented 1 month 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 month ago

Just for classification I do set the logger inside policy