default batch size

[jiujing23333]

Hi ranjaykrishna,

I love your excellent project, your code is really elegant.

I'm trying to reuse your code, but I met some problems with the batch size(by default that is 20), the batch size is too large for a single TITAN X, I have found for a single TITAN X, 2 is the max batch size. So do you use nn.dataparallel() or something else?

Any help is appreciated

Best Regards

[liluxuan1997]

I've also met the same problem. Looking forward to the reply.