search

wzzheng / OccSora

OccSora: 4D Occupancy Generation Models as World Simulators for Autonomous Driving
https://wzzheng.net/OccSora/
Apache License 2.0
132 stars 7 forks source link

Regarding memory usage issues #2

Closed jiangchaokang closed 3 months ago

jiangchaokang commented 3 months ago

Could you please tell me if it is possible to successfully train and evaluate OccSora using multiple (32) 3090 GPUs, even though I don't have a GPU with 80GB A100 memory?

LeningWang commented 3 months ago

You can compress with 12 frames instead of 32, so 24 GB GPU memory should be sufficient.