summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-18 18:57:26 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-18 18:57:26 +0200
commit84130a5e8cc1a0176295d23640ed2aaf5a92e557 (patch)
tree5a4e12224ebd16dbcadea1509fdaa7927413aefd
parentc4df0e0ae8f0980f509b09a7c2ba4014cceff2cc (diff)
Update wandb logger args
-rw-r--r--training/conf/callbacks/wandb/watch.yaml3
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