YanjieZe/3D-Diffusion-Policy

parallel training

Closed this issue · 1 comments

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

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