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

ok val but hacked away

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