diff options
| -rw-r--r-- | training/conf/callbacks/wandb/watch.yaml | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/training/conf/callbacks/wandb/watch.yaml b/training/conf/callbacks/wandb/watch.yaml index 511608c..bada03b 100644 --- a/training/conf/callbacks/wandb/watch.yaml +++ b/training/conf/callbacks/wandb/watch.yaml @@ -1,4 +1,5 @@ watch_model: _target_: callbacks.wandb_callbacks.WatchModel - log: all + log: gradients log_freq: 100 + log_graph: true |