From f5bf534f4227f2f482e722a3b2f2abac8bd62476 Mon Sep 17 00:00:00 2001
From: Patrick Esser <patrick.esser@gmx.net>
Date: Mon, 27 Jun 2022 07:42:32 +0000
Subject: [PATCH] ckpt every 5k

---
 .../txt2img-multinode-clip-encoder-f16-768-laion-hr.yaml      | 4 ++++
 1 file changed, 4 insertions(+)

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
index ab30a80..dc685cd 100644
--- 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