adapted sampling config for clip-encoder training

This commit is contained in:
rromb 2022-06-01 09:36:27 +02:00
parent abd9fb7af1
commit 634a591701

View file

@ -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