summaryrefslogtreecommitdiff
path: root/training/conf/callbacks
diff options
context:
space:
mode:
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