arobey1 / mbdg

MIT License
35 stars 4 forks source link

Why is the batch size so small? #1

Closed zinccat closed 2 years ago

zinccat commented 2 years ago

In code files that create dataloaders, I found that batchsize=1, I wonder is there any particular reason that you set this value so small? Thanks

arobey1 commented 2 years ago

This is the default batch size for MUNIT (see Appendix B of https://arxiv.org/pdf/1804.04732.pdf). When training classifiers, the default batch size should be larger.

zinccat commented 2 years ago

Thanks!