ckpt every 5k

codeaudit · Jun 27, 2022 · f5bf534 · f5bf534
1 parent 3809ba7
commit f5bf534
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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