r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

val loop

Browse source
This commit is contained in:
Patrick Esser 2022-05-30 22:16:58 +00:00 committed by root
parent ac1ca73d69
commit c7ea00aaf9
2 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
configs/stable-diffusion/dev_mn.yaml
View file

@ -93,8 +93,9 @@ data:
params: params:
size: 256 size: 256
# NOTE use enough shards to avoid empty validation loops in workers
validation: validation:
shards: '{231340..231349}.tar -' shards: '{231346..231349}.tar -'
shuffle: 0 shuffle: 0
image_key: jpg image_key: jpg
image_transforms: image_transforms:
@ -112,16 +113,16 @@ lightning:
image_logger: image_logger:
target: main.ImageLogger target: main.ImageLogger
params: params:
batch_frequency: 5000 # 5000 batch_frequency: 500 # 5000
max_images: 0 max_images: 8
increase_log_steps: False increase_log_steps: False
log_first_step: True log_first_step: False
trainer: trainer:
#replace_sampler_ddp: False #replace_sampler_ddp: False
benchmark: True benchmark: True
val_check_interval: 20000 # every 20k training steps val_check_interval: 1000 # every 20k training steps
num_sanity_val_steps: 0 num_sanity_val_steps: 0

9
configs/stable-diffusion/dev_mn_dummy.yaml
View file

@ -82,7 +82,7 @@ data:
train: train:
target: ldm.data.dummy.DummyData target: ldm.data.dummy.DummyData
params: params:
length: 10000 length: 20000
size: [256, 256, 3] size: [256, 256, 3]
validation: validation:
target: ldm.data.dummy.DummyData target: ldm.data.dummy.DummyData
@ -96,13 +96,14 @@ lightning:
image_logger: image_logger:
target: main.ImageLogger target: main.ImageLogger
params: params:
batch_frequency: 5000 # 5000 batch_frequency: 500 # 5000
max_images: 0 max_images: 8
increase_log_steps: False increase_log_steps: False
log_first_step: True log_first_step: False
trainer: trainer:
#replace_sampler_ddp: False #replace_sampler_ddp: False
benchmark: True benchmark: True
val_check_interval: 1000 # every 20k training steps
num_sanity_val_steps: 0 num_sanity_val_steps: 0