summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/wandb
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2023-09-02 01:53:55 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2023-09-02 01:53:55 +0200
commitcf42f987a34d4de0db10f733fd532f949dd9c278 (patch)
tree1420cc62d2792a3921272f4a5aa39e40ef7f7ce9 /training/conf/callbacks/wandb
parentabc2d60d69d115cdb34615d8bcb6c03ab6357141 (diff)
Update configs
Diffstat (limited to 'training/conf/callbacks/wandb')
-rw-r--r--training/conf/callbacks/wandb/captions.yaml4
-rw-r--r--training/conf/callbacks/wandb/checkpoints.yaml2
-rw-r--r--training/conf/callbacks/wandb/config.yaml2
-rw-r--r--training/conf/callbacks/wandb/predictions.yaml3
-rw-r--r--training/conf/callbacks/wandb/watch.yaml2
5 files changed, 7 insertions, 6 deletions
diff --git a/training/conf/callbacks/wandb/captions.yaml b/training/conf/callbacks/wandb/captions.yaml
new file mode 100644
index 0000000..3215a90
--- /dev/null
+++ b/training/conf/callbacks/wandb/captions.yaml
@@ -0,0 +1,4 @@
+log_text_predictions:
+ _target_: callbacks.wandb.ImageToCaptionLogger
+ num_samples: 8
+ on_train: true
diff --git a/training/conf/callbacks/wandb/checkpoints.yaml b/training/conf/callbacks/wandb/checkpoints.yaml
index a4a16ff..b9a3fd7 100644
--- a/training/conf/callbacks/wandb/checkpoints.yaml
+++ b/training/conf/callbacks/wandb/checkpoints.yaml
@@ -1,4 +1,4 @@
upload_ckpts_as_artifact:
- _target_: callbacks.wandb_callbacks.UploadCheckpointsAsArtifact
+ _target_: callbacks.wandb.UploadCheckpointsAsArtifact
ckpt_dir: checkpoints/
upload_best_only: true
diff --git a/training/conf/callbacks/wandb/config.yaml b/training/conf/callbacks/wandb/config.yaml
index 747a7c6..d841e94 100644
--- a/training/conf/callbacks/wandb/config.yaml
+++ b/training/conf/callbacks/wandb/config.yaml
@@ -1,2 +1,2 @@
upload_code_as_artifact:
- _target_: callbacks.wandb_callbacks.UploadConfigAsArtifact
+ _target_: callbacks.wandb.UploadConfigAsArtifact
diff --git a/training/conf/callbacks/wandb/predictions.yaml b/training/conf/callbacks/wandb/predictions.yaml
deleted file mode 100644
index 573fa96..0000000
--- a/training/conf/callbacks/wandb/predictions.yaml
+++ /dev/null
@@ -1,3 +0,0 @@
-log_text_predictions:
- _target_: callbacks.wandb_callbacks.LogTextPredictions
- num_samples: 8
diff --git a/training/conf/callbacks/wandb/watch.yaml b/training/conf/callbacks/wandb/watch.yaml
index 660ae47..1f60978 100644
--- a/training/conf/callbacks/wandb/watch.yaml
+++ b/training/conf/callbacks/wandb/watch.yaml
@@ -1,5 +1,5 @@
watch_model:
- _target_: callbacks.wandb_callbacks.WatchModel
+ _target_: callbacks.wandb.WatchModel
log_params: gradients
log_freq: 100
log_graph: true