Wandb.watch in accelerate library

[muellerzr]

@morgan

It might still work as long as its called before you start logging in your script

For code of what @morgan means, it’d look something like this:

accelerate = Accelerator(log_with="wandb")

accelerate.init_trackers("my_projectname")
wandb.watch()