code: def on_train_begin(self, **kwargs):
          "Call watch method to log model topology, gradients & weights"
  
          # Set self.best, method inherited from "TrackerCallback" by "SaveModelCallback"
          super().on_train_begin()
  
          # Ensure we don't call "watch" multiple times
          if not WandbCallback.watch_called:
              WandbCallback.watch_called = True
  
              # Logs model topology and optionally gradients and weights
              wandb.watch(self.learn.model, log=self.logf on_train_begin(self, **kwargs):
          "Call watch method to log model topology, gradients & weights"
  
          # Set self.best, method inherited from "TrackerCallback" by "SaveModelCallback"
          super().on_train_begin()
  
          # Ensure we don't call "watch" multiple times
          if not WandbCallback.watch_called:
              WandbCallback.watch_called = True
  
              # Logs model topology and optionally gradients and weights
              wandb.watch(self.learn.model, log=self.log)
docs: Call watch method to log model topology, gradients & weights