diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-08 22:26:15 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-08 22:26:15 +0200 |
| commit | 72befa6d8cc4c7ecf698512a97424641ee81725a (patch) | |
| tree | af2ad5662eb8385c51c33c00761bd6f6bf29d8ee /training/conf/callbacks | |
| parent | 544d37d9c3b8b57798d57988d6f000c19ab6b074 (diff) | |
Move callbacks to training folder, refactor
Diffstat (limited to 'training/conf/callbacks')
| -rw-r--r-- | training/conf/callbacks/wandb.yaml | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/training/conf/callbacks/wandb.yaml b/training/conf/callbacks/wandb.yaml index 2d56bfa..6eedb71 100644 --- a/training/conf/callbacks/wandb.yaml +++ b/training/conf/callbacks/wandb.yaml @@ -2,19 +2,19 @@ defaults: - default.yaml watch_model: - _target_: text_recognizer.callbacks.wandb_callbacks.WatchModel + _target_: callbacks.wandb_callbacks.WatchModel log: "all" log_freq: 100 upload_code_as_artifact: - _target_: text_recognizer.callbacks.wandb_callbacks.UploadCodeAsArtifact + _target_: callbacks.wandb_callbacks.UploadCodeAsArtifact project_dir: ${work_dir}/text_recognizer upload_ckpts_as_artifact: - _target_: text_recognizer.callbacks.wandb_callbacks.UploadCheckpointsAsArtifact + _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact ckpt_dir: "checkpoints/" upload_best_only: True log_text_predictions: - _target_: text_recognizer.callbacks.wandb_callbacks.LogTextPredictions + _target_: callbacks.wandb_callbacks.LogTextPredictions num_samples: 8 |