search

AIR-THU / DAIR-V2X-Seq

137 stars 15 forks source link

multi-GPU training #15

Closed Wansit99 closed 9 months ago

Wansit99 commented 10 months ago

Thank for your great work! Could HIVT support multi-GPU training?

HzRynn commented 10 months ago

Yes! Convienently set --gpus in train.py.

Wansit99 commented 10 months ago

Thanks！