antioom
This commit is contained in:
parent
9981c7209c
commit
cace7f06aa
1 changed files with 1 additions and 1 deletions
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue