search

YanjieZe / 3D-Diffusion-Policy

[RSS 2024] 3D Diffusion Policy: Generalizable Visuomotor Policy Learning via Simple 3D Representations
https://3d-diffusion-policy.github.io
MIT License
514 stars 50 forks source link

parallel training #70

Closed kangkang2626 closed 1 month ago

kangkang2626 commented 1 month ago

hi, is it possible to train dp3 in parallel using multiple gpu's?

YanjieZe commented 1 month ago

I think it is possible. The modules of DP3 are all vanilla pytorch networks. But I have not tried that actually.