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

make it ready

This commit is contained in:
rromb 2022-06-10 12:27:35 +02:00
parent c02c06b778
commit 34f9f3867e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-256-pretraining.yaml
View File

@ -70,7 +70,7 @@ data:
target: ldm.data.laion.WebDataModuleFromConfig
params:
tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/"
batch_size: 50 # TODO: max out
batch_size: 55
num_workers: 4
multinode: True
min_size: 256 # TODO: experiment. Note: for 2B, images are stored at max 384 resolution
@ -124,4 +124,4 @@ lightning:
benchmark: True
val_check_interval: 5000000 # really sorry
num_sanity_val_steps: 0
accumulate_grad_batches: 2 # TODO: want accumulate on? --> wait for final batch-size
accumulate_grad_batches: 2