diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-30 23:08:15 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-30 23:08:15 +0200 |
commit | 4f0469f755507b15dae65510b2000a0ce077b423 (patch) | |
tree | c302d6071799756b591ac19710025075a5725476 /training/conf | |
parent | af1fccaa154a4455ea6b5d6c5b1da1e1e427edba (diff) |
Add new variables to w&b callbacks
Diffstat (limited to 'training/conf')
-rw-r--r-- | training/conf/callbacks/wandb_htr_predictions.yaml | 1 | ||||
-rw-r--r-- | training/conf/callbacks/wandb_image_reconstructions.yaml | 2 |
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 |