The training GPU memory requirement

chenyilun95 / DSGN2

DSGN++: Exploiting Visual-Spatial Relation for Stereo-based 3D Detectors (T-PAMI 2022)

Apache License 2.0

79 stars 8 forks source link

The training GPU memory requirement #8

Closed wangyd-0312 closed 1 year ago

wangyd-0312 commented 1 year ago

Hello! Thank you for your awesome work. I notice that your previous work ,DSGN, required GPU memory to close to 29G. How about DSGN2? How much memory I need to train or test this model? Looking for your replay. Best wishes.

chenyilun95 commented 1 year ago

The models are trained with 24 GB GPU memory.