How to modify the batch_size?

[hengyu-wang]

Congratulations on your excellent work！
When I train other data sets, I found that modifying the batch_size in init.py will cause an error. Can you tell me how to modify the batch_size to train normally?

Thanks!