From cf42f987a34d4de0db10f733fd532f949dd9c278 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 2 Sep 2023 01:53:55 +0200 Subject: Update configs --- training/conf/callbacks/htr.yaml | 2 +- training/conf/callbacks/wandb/captions.yaml | 4 ++++ training/conf/callbacks/wandb/checkpoints.yaml | 2 +- training/conf/callbacks/wandb/config.yaml | 2 +- training/conf/callbacks/wandb/predictions.yaml | 3 --- training/conf/callbacks/wandb/watch.yaml | 2 +- 6 files changed, 8 insertions(+), 7 deletions(-) create mode 100644 training/conf/callbacks/wandb/captions.yaml delete mode 100644 training/conf/callbacks/wandb/predictions.yaml (limited to 'training/conf/callbacks') diff --git a/training/conf/callbacks/htr.yaml b/training/conf/callbacks/htr.yaml index ebcb0fa..5bd9b12 100644 --- a/training/conf/callbacks/htr.yaml +++ b/training/conf/callbacks/htr.yaml @@ -1,3 +1,3 @@ defaults: - default - - wandb/predictions + - wandb/captions diff --git a/training/conf/callbacks/wandb/captions.yaml b/training/conf/callbacks/wandb/captions.yaml new file mode 100644 index 0000000..3215a90 --- /dev/null +++ b/training/conf/callbacks/wandb/captions.yaml @@ -0,0 +1,4 @@ +log_text_predictions: + _target_: callbacks.wandb.ImageToCaptionLogger + num_samples: 8 + on_train: true diff --git a/training/conf/callbacks/wandb/checkpoints.yaml b/training/conf/callbacks/wandb/checkpoints.yaml index a4a16ff..b9a3fd7 100644 --- a/training/conf/callbacks/wandb/checkpoints.yaml +++ b/training/conf/callbacks/wandb/checkpoints.yaml @@ -1,4 +1,4 @@ upload_ckpts_as_artifact: - _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact + _target_: callbacks.wandb.UploadCheckpointsAsArtifact ckpt_dir: checkpoints/ upload_best_only: true diff --git a/training/conf/callbacks/wandb/config.yaml b/training/conf/callbacks/wandb/config.yaml index 747a7c6..d841e94 100644 --- a/training/conf/callbacks/wandb/config.yaml +++ b/training/conf/callbacks/wandb/config.yaml @@ -1,2 +1,2 @@ upload_code_as_artifact: - _target_: callbacks.wandb_callbacks.UploadConfigAsArtifact + _target_: callbacks.wandb.UploadConfigAsArtifact diff --git a/training/conf/callbacks/wandb/predictions.yaml b/training/conf/callbacks/wandb/predictions.yaml deleted file mode 100644 index 573fa96..0000000 --- a/training/conf/callbacks/wandb/predictions.yaml +++ /dev/null @@ -1,3 +0,0 @@ -log_text_predictions: - _target_: callbacks.wandb_callbacks.LogTextPredictions - num_samples: 8 diff --git a/training/conf/callbacks/wandb/watch.yaml b/training/conf/callbacks/wandb/watch.yaml index 660ae47..1f60978 100644 --- a/training/conf/callbacks/wandb/watch.yaml +++ b/training/conf/callbacks/wandb/watch.yaml @@ -1,5 +1,5 @@ watch_model: - _target_: callbacks.wandb_callbacks.WatchModel + _target_: callbacks.wandb.WatchModel log_params: gradients log_freq: 100 log_graph: true -- cgit v1.2.3-70-g09d2