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

ok val but hacked away

This commit is contained in:
rromb 2022-05-31 16:12:02 +02:00
parent c537709d4d
commit 5fd73effd3
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
View File

@ -90,19 +90,19 @@ data:
params:
size: 256
# # NOTE use enough shards to avoid empty validation loops in workers
# validation:
# shards: '{231318..231349}.tar -'
# shuffle: 0
# image_key: jpg
# image_transforms:
# - target: torchvision.transforms.Resize
# params:
# size: 256
# interpolation: 3
# - target: torchvision.transforms.CenterCrop
# params:
# size: 256
# NOTE use enough shards to avoid empty validation loops in workers
validation:
shards: '{231318..231349}.tar -'
shuffle: 0
image_key: jpg
image_transforms:
- target: torchvision.transforms.Resize
params:
size: 256
interpolation: 3
- target: torchvision.transforms.CenterCrop
params:
size: 256
lightning:
@ -119,6 +119,6 @@ lightning:
trainer:
#replace_sampler_ddp: False
benchmark: True
#val_check_interval: 50000
#num_sanity_val_steps: 0
val_check_interval: 5000000 # really sorry
num_sanity_val_steps: 0
accumulate_grad_batches: 2