SCNN icon indicating copy to clipboard operation
SCNN copied to clipboard

Training problem

Open BoSeal opened this issue 6 years ago • 3 comments

When I train the lane dataset, I use the hardware configuration: one Tesla K20 GPU with 5G memory. This causes OOM!To solve this problem, I reduce the batch size to 4 ,but still happen“OOM”. So I don't know if I'm going to keep reducing the batch size? If I continue to reduce the batch size, I worry that it will affect the accuracy of the results and not reach the accuracy obtained in your paper. Can you give me some advice?

BoSeal avatar Apr 28 '19 08:04 BoSeal

@BoSeal Reducing batchsize would harm the performance of BN. One possible solution to this is to use frozen BN and smaller batchsize (like 2) as done in some semantic segmentation works.

XingangPan avatar Apr 29 '19 11:04 XingangPan

Thanks very much!

BoSeal avatar Apr 30 '19 08:04 BoSeal

Why does reducing batchsize harm the performance of BN? What is BN?

Thanks in advance

OrkunYilmaz avatar May 02 '19 13:05 OrkunYilmaz