zgojcic / Rigid3DSceneFlow

[CVPR 2021, Oral] "Weakly Supervised Learning of Rigid 3D Scene Flow"
137 stars 18 forks source link

Multi-GPU Train #19

Open YQChen8 opened 1 year ago

YQChen8 commented 1 year ago

Thank you for your excellent work. I would like to inquire whether this code supports multiple GPUs. If it does, I would like to know where to enable it. If it doesn't, could you please provide some modification suggestions? Thank you!