search

xy-guo / LIGA-Stereo

Code for LIGA-Stereo Detector, ICCV'21
Apache License 2.0
90 stars 18 forks source link

Batch size > 1 on single GPU #8

Open ngoductuanlhp opened 2 years ago

ngoductuanlhp commented 2 years ago

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 commented 2 years ago

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.