From 969e1d5e179d9c42ffae0c9b12c9bd3be6091360 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 4 Jul 2021 22:58:07 +0200
Subject: Add wandb callbacks

---
 training/conf/callbacks/wandb.yaml | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 training/conf/callbacks/wandb.yaml

(limited to 'training/conf/callbacks/wandb.yaml')

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
-- 
cgit v1.2.3-70-g09d2