From bd4bd443f339e95007bfdabf3e060db720f4d4b9 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 3 Aug 2021 18:18:48 +0200 Subject: Training working, multiple bug fixes --- training/conf/callbacks/checkpoint.yaml | 2 +- training/conf/callbacks/wandb/checkpoints.yaml | 4 ---- training/conf/callbacks/wandb/code.yaml | 3 --- training/conf/callbacks/wandb/image_reconstructions.yaml | 0 training/conf/callbacks/wandb/ocr_predictions.yaml | 3 --- training/conf/callbacks/wandb/watch.yaml | 4 ---- training/conf/callbacks/wandb_checkpoints.yaml | 4 ++++ training/conf/callbacks/wandb_code.yaml | 3 +++ training/conf/callbacks/wandb_image_reconstructions.yaml | 0 training/conf/callbacks/wandb_ocr.yaml | 8 ++++---- training/conf/callbacks/wandb_ocr_predictions.yaml | 3 +++ training/conf/callbacks/wandb_watch.yaml | 4 ++++ 12 files changed, 19 insertions(+), 19 deletions(-) delete mode 100644 training/conf/callbacks/wandb/checkpoints.yaml delete mode 100644 training/conf/callbacks/wandb/code.yaml delete mode 100644 training/conf/callbacks/wandb/image_reconstructions.yaml delete mode 100644 training/conf/callbacks/wandb/ocr_predictions.yaml delete mode 100644 training/conf/callbacks/wandb/watch.yaml create mode 100644 training/conf/callbacks/wandb_checkpoints.yaml create mode 100644 training/conf/callbacks/wandb_code.yaml create mode 100644 training/conf/callbacks/wandb_image_reconstructions.yaml create mode 100644 training/conf/callbacks/wandb_ocr_predictions.yaml create mode 100644 training/conf/callbacks/wandb_watch.yaml (limited to 'training/conf/callbacks') diff --git a/training/conf/callbacks/checkpoint.yaml b/training/conf/callbacks/checkpoint.yaml index db34cb1..b4101d8 100644 --- a/training/conf/callbacks/checkpoint.yaml +++ b/training/conf/callbacks/checkpoint.yaml @@ -6,4 +6,4 @@ model_checkpoint: mode: min # can be "max" or "min" verbose: false dirpath: checkpoints/ - filename: {epoch:02d} + filename: "{epoch:02d}" diff --git a/training/conf/callbacks/wandb/checkpoints.yaml b/training/conf/callbacks/wandb/checkpoints.yaml deleted file mode 100644 index a4a16ff..0000000 --- a/training/conf/callbacks/wandb/checkpoints.yaml +++ /dev/null @@ -1,4 +0,0 @@ -upload_ckpts_as_artifact: - _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact - ckpt_dir: checkpoints/ - upload_best_only: true diff --git a/training/conf/callbacks/wandb/code.yaml b/training/conf/callbacks/wandb/code.yaml deleted file mode 100644 index 35f6ea3..0000000 --- a/training/conf/callbacks/wandb/code.yaml +++ /dev/null @@ -1,3 +0,0 @@ -upload_code_as_artifact: - _target_: callbacks.wandb_callbacks.UploadCodeAsArtifact - project_dir: ${work_dir}/text_recognizer diff --git a/training/conf/callbacks/wandb/image_reconstructions.yaml b/training/conf/callbacks/wandb/image_reconstructions.yaml deleted file mode 100644 index e69de29..0000000 diff --git a/training/conf/callbacks/wandb/ocr_predictions.yaml b/training/conf/callbacks/wandb/ocr_predictions.yaml deleted file mode 100644 index 573fa96..0000000 --- a/training/conf/callbacks/wandb/ocr_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 deleted file mode 100644 index 511608c..0000000 --- a/training/conf/callbacks/wandb/watch.yaml +++ /dev/null @@ -1,4 +0,0 @@ -watch_model: - _target_: callbacks.wandb_callbacks.WatchModel - log: all - log_freq: 100 diff --git a/training/conf/callbacks/wandb_checkpoints.yaml b/training/conf/callbacks/wandb_checkpoints.yaml new file mode 100644 index 0000000..a4a16ff --- /dev/null +++ b/training/conf/callbacks/wandb_checkpoints.yaml @@ -0,0 +1,4 @@ +upload_ckpts_as_artifact: + _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact + ckpt_dir: checkpoints/ + upload_best_only: true diff --git a/training/conf/callbacks/wandb_code.yaml b/training/conf/callbacks/wandb_code.yaml new file mode 100644 index 0000000..35f6ea3 --- /dev/null +++ b/training/conf/callbacks/wandb_code.yaml @@ -0,0 +1,3 @@ +upload_code_as_artifact: + _target_: callbacks.wandb_callbacks.UploadCodeAsArtifact + project_dir: ${work_dir}/text_recognizer diff --git a/training/conf/callbacks/wandb_image_reconstructions.yaml b/training/conf/callbacks/wandb_image_reconstructions.yaml new file mode 100644 index 0000000..e69de29 diff --git a/training/conf/callbacks/wandb_ocr.yaml b/training/conf/callbacks/wandb_ocr.yaml index efa3dda..9c9a6da 100644 --- a/training/conf/callbacks/wandb_ocr.yaml +++ b/training/conf/callbacks/wandb_ocr.yaml @@ -1,6 +1,6 @@ defaults: - default - - wandb/watch - - wandb/code - - wandb/checkpoints - - wandb/ocr_predictions + - wandb_watch + - wandb_code + - wandb_checkpoints + - wandb_ocr_predictions diff --git a/training/conf/callbacks/wandb_ocr_predictions.yaml b/training/conf/callbacks/wandb_ocr_predictions.yaml new file mode 100644 index 0000000..573fa96 --- /dev/null +++ b/training/conf/callbacks/wandb_ocr_predictions.yaml @@ -0,0 +1,3 @@ +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 new file mode 100644 index 0000000..511608c --- /dev/null +++ b/training/conf/callbacks/wandb_watch.yaml @@ -0,0 +1,4 @@ +watch_model: + _target_: callbacks.wandb_callbacks.WatchModel + log: all + log_freq: 100 -- cgit v1.2.3-70-g09d2