watch_model:
  _target_: callbacks.wandb_callbacks.WatchModel
  log_params: gradients
  log_freq: 100
  log_graph: true