From 84130a5e8cc1a0176295d23640ed2aaf5a92e557 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 18 Sep 2022 18:57:26 +0200 Subject: Update wandb logger args --- training/conf/callbacks/wandb/watch.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 -- cgit v1.2.3-70-g09d2