bs and unused params
This commit is contained in:
parent
12d2aaf50c
commit
9981c7209c
1 changed files with 3 additions and 1 deletions
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue