wandb_logger.watch(log_graph=True/False) not working

wandb_logger.watch(model, log_graph=False) is not working.

Error : watch() got an unexpected keyword argument ‘log_graph’

hey @Kirtikumar_Pandya

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