Batch size > 1 on single GPU

[ngoductuanlhp]

First, thank you for your great work and code.

I saw in your code that you force the batch_size_per_gpu = 1. What's the reason for this config? If I want to train a larger batch size on a single GPU, which parts should I modify?

Look forward to your answer. Thanks.

[xy-guo]

The 3D CNN consumes a lot of memory, it is hard to train with larger batch size. A possible solution is to reduce the number of channels or layers, but it will affect the performance.