MapTR on a single GPU??

hustvl / MapTR

[ICLR'23 Spotlight] MapTR: Structured Modeling and Learning for Online Vectorized HD Map Construction

MIT License

984 stars 152 forks source link

Open mohamedsamirx opened 10 months ago

mohamedsamirx commented 10 months ago

how to train and evaluate maptr or maptrv2 on a single gpu as all the code use torch.distributed ??

eraofelix commented 8 months ago

you can train like this: python tools/train.py ./projects/configs/maptrv2/maptrv2_nusc_r50_24ep.py