search

rhett-chen / graspness_implementation

My implementation of Graspnet Graspness.
Other
139 stars 33 forks source link

GPU memory size #17

Closed Jerry-robot closed 1 year ago

Jerry-robot commented 1 year ago

Hello, how much video memory is needed to train the data with this model? Thank you very much!

rhett-chen commented 1 year ago

Hi, you can train on single 3080 GPU with bs=2, ~10G GPU memory cost