summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/wandb.yaml
blob: 6eedb715b2bedae4c3bdf6c0bada1a464ca895d4 (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