aimagelab / VATr

MIT License
74 stars 4 forks source link

is batch-size set to 1 during training? #18

Closed yuyukuy534 closed 1 year ago

yuyukuy534 commented 1 year ago

Hello, is batch-size set to 1 during training?

vittoriopippi commented 1 year ago

No https://github.com/aimagelab/VATr/blob/6ef5a91b541c92b448191a7f424e29db6e117e3d/train.py#L31C28-L31C28