wandb_logger.watch(model, log_graph=False) is not working.
Error : watch() got an unexpected keyword argument ‘log_graph’
I believe it might have been added to a new version of Wandb. You might want to update the package.
Also, we have moved the discussions to GitHub Discussions. You might want to check that out instead to get a quick response. The forums will be marked read-only after some time.
Thank you