diff options
Diffstat (limited to 'training')
| -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 |