def on_train_begin(self, **kwargs):
    'Call watch method to log model topology, gradients & weights'
    super().on_train_begin()
    if (not WandbCallback._watch_called):
        WandbCallback._watch_called = True
        wandb.watch(self.learn.model, log=self.log)
Call watch method to log model topology, gradients & weights
def on_train_begin(self, **kwargs):
    
    super().on_train_begin()
    if (not WandbCallback._watch_called):
        WandbCallback._watch_called = True
        wandb.watch(self.learn.model, log=self.log)