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
1 changed files with 3 additions and 1 deletions

View File

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