ckpt every 5k

2022-06-27 07:42:32 +00:00 · 2022-06-27 07:42:32 +00:00 · f5bf534f42
commit f5bf534f42
parent 3809ba7046
1 changed files with 4 additions and 0 deletions
--- a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768-laion-hr.yaml
+++ b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768-laion-hr.yaml
@ -105,6 +105,10 @@ data:
 lightning:
  find_unused_parameters: False

+  modelcheckpoint:
+    params:
+      every_n_train_steps: 5000
+
  callbacks:
    image_logger:
      target: main.ImageLogger