bs and unused params

This commit is contained in:
Patrick Esser 2022-06-17 23:20:03 +00:00 committed by root
parent 12d2aaf50c
commit 9981c7209c

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: 10 batch_size: 8
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
@ -104,6 +104,8 @@ data:
lightning: lightning:
find_unused_parameters: False
callbacks: callbacks:
image_logger: image_logger:
target: main.ImageLogger target: main.ImageLogger