adapted sampling config for clip-encoder training
This commit is contained in:
parent
abd9fb7af1
commit
634a591701
1 changed files with 8 additions and 3 deletions
|
@ -110,11 +110,16 @@ lightning:
|
||||||
image_logger:
|
image_logger:
|
||||||
target: main.ImageLogger
|
target: main.ImageLogger
|
||||||
params:
|
params:
|
||||||
batch_frequency: 5000
|
batch_frequency: 50
|
||||||
max_images: 8
|
max_images: 4
|
||||||
increase_log_steps: False
|
increase_log_steps: False
|
||||||
log_first_step: 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:
|
trainer:
|
||||||
#replace_sampler_ddp: False
|
#replace_sampler_ddp: False
|
||||||
|
|
Loading…
Reference in a new issue