aimagelab / VATr

MIT License
72 stars 4 forks source link

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

Closed yuyukuy534 closed 11 months ago

yuyukuy534 commented 11 months ago

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

vittoriopippi commented 11 months ago

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