adapted sampling config for clip-encoder training

2022-06-01 09:36:27 +02:00 · 2022-06-01 09:36:27 +02:00 · 634a591701
parent abd9fb7af1
commit 634a591701
1 changed files with 8 additions and 3 deletions
--- a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
+++ b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
@ -110,11 +110,16 @@ lightning:
    image_logger:
      target: main.ImageLogger
      params:
-        batch_frequency: 5000
-        max_images: 8
+        batch_frequency: 50
+        max_images: 4
        increase_log_steps: False
        log_first_step: False
-
+        log_images_kwargs:
+          plot_progressive_rows: False
+          plot_diffusion_rows: False
+          N: 4
+          unconditional_guidance_scale: 3.0
+          unconditional_guidance_label: ""

  trainer:
    #replace_sampler_ddp: False