search

qcraftai / simtrack

Exploring Simple 3D Multi-Object Tracking for Autonomous Driving (ICCV 2021)
Other
171 stars 31 forks source link

12GB GPU memory is not enough for training with default config #8

Closed qiyancos closed 2 years ago

qiyancos commented 2 years ago

I have a gpu card with only 12GB video memory. The process failed with out-of-memory error when I tried to run with default config. What should I change in the default config to run your model normally?

xiaodongyang commented 2 years ago

You may try with a smaller batch size and adjust the learning rate. AMP also helps.