search

mosturm / SynCellFactory

2 stars 0 forks source link

How to change batch_size from 4 to other number? #1

Closed istan-ding closed 1 month ago

istan-ding commented 2 months ago

How to change batch_size from 4 to other number?

mosturm commented 1 month ago

Hi Istan,

You can change the batch size for training and validation in the DataLoaders. They are located in SynCellFactory/ControlNet/cldm/cldm.py, starting from line 449. The batch size variable is called bs.

Cheers, Mo

istan-ding commented 1 month ago

Thank you very much for your guidance!