search

Banconxuan / RTM3D

The official PyTorch Implementation of RTM3D and KM3D for Monocular 3D Object Detection
MIT License
454 stars 85 forks source link

Training with multiple GPUs #26

Closed mertmerci closed 3 years ago

mertmerci commented 3 years ago

Hello,

Is it possible to run the training using two GPUs? If yes, how should I adapt the code to do it?

Thanks in advance.

Banconxuan commented 3 years ago

add --gpus 0,1 in the training parameters