Hi~ I wonder why not use a bigger batch_size in "train.py" <img width="739" alt="Screenshot 2022-12-08 at 15 25 33" src="https://user-images.githubusercontent.com/88081482/206385249-859222fb-0dd1-48c9-9e7c-37696d03d4b2.png">
