OpenDriveLab / UniAD

[CVPR'23 Best Paper Award] Planning-oriented Autonomous Driving
Apache License 2.0
3.11k stars 335 forks source link

Encounted OOM while trying to train the stage1 model #163

Open stevenk12 opened 5 months ago

stevenk12 commented 5 months ago

I use the script 'uni_dist_train.sh' and try to train the stage1(track and map) model, but I encounter the error of 'OUT OF MEMORY'. I use four 3090 GPUs, the total memory is 96GB. Recording to your words in 'Eval and Train', the first stage needs about 50GB memory. So what's the problem? Can u help me. Thanks!