princeton-vl / RAFT-Stereo

MIT License
721 stars 139 forks source link

Training GPU memory #92

Open TomTomTommi opened 1 year ago

TomTomTommi commented 1 year ago

Hi, I trained the model using the given command on sceneflow dataset on 1 A6000. I found the GPU memory cost is about 39G. Just a double-check with you: is it correct? Since I saw you claimed the GPU is 2 A6000.