This commit is contained in:
Patrick Esser 2022-06-18 07:30:59 +00:00 committed by root
parent 9981c7209c
commit cace7f06aa

View file

@ -71,7 +71,7 @@ data:
target: ldm.data.laion.WebDataModuleFromConfig target: ldm.data.laion.WebDataModuleFromConfig
params: params:
tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/" tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/"
batch_size: 8 batch_size: 6
num_workers: 4 num_workers: 4
multinode: True multinode: True
min_size: 384 # TODO: experiment. Note: for 2B, images are stored at max 384 resolution min_size: 384 # TODO: experiment. Note: for 2B, images are stored at max 384 resolution