search

yuantianyuan01 / StreamMapNet

GNU General Public License v3.0
202 stars 17 forks source link

Single GPU training #4

Closed alfredgu001324 closed 1 year ago

alfredgu001324 commented 1 year ago

Hi, thank for the great work.

I am trying to reproduce your result using just one single GPU. But the following error occurs: RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.

by running this command: python tools/train.py /home/guxunjia/project/StreamMapNet/plugin/configs/nusc_baseline_480_60x30_30e.py --deterministic --no-validate

Zhenghao97 commented 10 months ago

You should start the training by torch.distributed.launch even if you only use one gpu. What's more, according to the comment code of this repo, the bevformer backbone only support distributed training.