diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-04 22:58:07 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-04 22:58:07 +0200 |
| commit | 969e1d5e179d9c42ffae0c9b12c9bd3be6091360 (patch) | |
| tree | ab849c38bc9b863afad85fd04d6f618031000e6f /training/conf/callbacks/wandb.yaml | |
| parent | 4da7a2c812221d56a430b35139ac40b23fa76f77 (diff) | |
Add wandb callbacks
Diffstat (limited to 'training/conf/callbacks/wandb.yaml')
| -rw-r--r-- | training/conf/callbacks/wandb.yaml | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/training/conf/callbacks/wandb.yaml b/training/conf/callbacks/wandb.yaml new file mode 100644 index 0000000..2d56bfa --- /dev/null +++ b/training/conf/callbacks/wandb.yaml @@ -0,0 +1,20 @@ +defaults: + - default.yaml + +watch_model: + _target_: text_recognizer.callbacks.wandb_callbacks.WatchModel + log: "all" + log_freq: 100 + +upload_code_as_artifact: + _target_: text_recognizer.callbacks.wandb_callbacks.UploadCodeAsArtifact + project_dir: ${work_dir}/text_recognizer + +upload_ckpts_as_artifact: + _target_: text_recognizer.callbacks.wandb_callbacks.UploadCheckpointsAsArtifact + ckpt_dir: "checkpoints/" + upload_best_only: True + +log_text_predictions: + _target_: text_recognizer.callbacks.wandb_callbacks.LogTextPredictions + num_samples: 8 |