shinya7y / UniverseNet

USB: Universal-Scale Object Detection Benchmark (BMVC 2022)
Apache License 2.0
422 stars 54 forks source link

Hardware requirements for trainning Waymo Open Dataset #25

Closed anhhna closed 3 years ago

anhhna commented 3 years ago

image

In the README said that "A machine with 208-416 GB of CPU memory is needed for full training as of MMDetection v2.0.", CPU memory that you means in this situation is RAM?, so what's the requirements of GPU, CPU, RAM, Hard disk for training this dataset, thank you so much for answering this quetion!

shinya7y commented 3 years ago

CPU memory means RAM. The RAM usage could be reduced by changing code and annotation format.

shinya7y commented 3 years ago

You can try f0train (1/10 size dataset), which requires less RAM. However, training on one RTX 3070 will take a very long time. I recommend smaller datasets.

anhhna commented 3 years ago

ok thank you so much, I will try it, I closed the issue