crywang/RFM

BatchSize settings

Closed this issue · 1 comments

reallm commented

Hi,

Great work~ Thanks.

I have a question: Why the batchsize is only 16 in paper experiments, what will happen if I set it as 128, 256 or larger?

crywang commented

We use 2080TI for training, and the batch size is limited by insufficient graphic memory. I guess the training will be more stable if you set up a larger batch size.