summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/wandb_image_reconstructions.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/callbacks/wandb_image_reconstructions.yaml')
-rw-r--r--training/conf/callbacks/wandb_image_reconstructions.yaml2
1 files changed, 2 insertions, 0 deletions
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