summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/wandb.yaml
blob: 0017e11bd8fb02fa123583ceea29c3869331c42e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
defaults:
  - default.yaml

watch_model:
  _target_: callbacks.wandb_callbacks.WatchModel
  log: all
  log_freq: 100

upload_code_as_artifact:
  _target_: callbacks.wandb_callbacks.UploadCodeAsArtifact
  project_dir: ${work_dir}/text_recognizer

upload_ckpts_as_artifact:
  _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact
  ckpt_dir: checkpoints/
  upload_best_only: true

log_text_predictions:
  _target_: callbacks.wandb_callbacks.LogTextPredictions
  num_samples: 8