summaryrefslogtreecommitdiff
path: root/training/conf/callbacks
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-30 23:08:15 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-30 23:08:15 +0200
commit4f0469f755507b15dae65510b2000a0ce077b423 (patch)
treec302d6071799756b591ac19710025075a5725476 /training/conf/callbacks
parentaf1fccaa154a4455ea6b5d6c5b1da1e1e427edba (diff)
Add new variables to w&b callbacks
Diffstat (limited to 'training/conf/callbacks')
-rw-r--r--training/conf/callbacks/wandb_htr_predictions.yaml1
-rw-r--r--training/conf/callbacks/wandb_image_reconstructions.yaml2
2 files changed, 3 insertions, 0 deletions
diff --git a/training/conf/callbacks/wandb_htr_predictions.yaml b/training/conf/callbacks/wandb_htr_predictions.yaml
index 573fa96..589e7e0 100644
--- a/training/conf/callbacks/wandb_htr_predictions.yaml
+++ b/training/conf/callbacks/wandb_htr_predictions.yaml
@@ -1,3 +1,4 @@
log_text_predictions:
_target_: callbacks.wandb_callbacks.LogTextPredictions
num_samples: 8
+ log_train: true
diff --git a/training/conf/callbacks/wandb_image_reconstructions.yaml b/training/conf/callbacks/wandb_image_reconstructions.yaml
index 6cc4ada..fabfe31 100644
--- a/training/conf/callbacks/wandb_image_reconstructions.yaml
+++ b/training/conf/callbacks/wandb_image_reconstructions.yaml
@@ -1,3 +1,5 @@
log_image_reconstruction:
_target_: callbacks.wandb_callbacks.LogReconstuctedImages
num_samples: 8
+ log_train: true
+ use_sigmoid: true