summaryrefslogtreecommitdiff
path: root/training/conf/callbacks/wandb/captions.yaml
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/captions.yaml
parentabc2d60d69d115cdb34615d8bcb6c03ab6357141 (diff)
Update configs
Diffstat (limited to 'training/conf/callbacks/wandb/captions.yaml')
-rw-r--r--training/conf/callbacks/wandb/captions.yaml4
1 files changed, 4 insertions, 0 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