robobase-org / robobase

27 stars 4 forks source link

Inquiry regarding multi-GPU support #2

Open rl-max opened 3 weeks ago

rl-max commented 3 weeks ago

Hello, thanks for sharing a nice framework for robot learning.

I was using robobase for training behavior cloning policy in BiGym and came across difficulty in accelerating the training due to the use of only single-GPU in the code.

Do the authors plan to release multi-GPU support feature in robobase (eg., DistributedDataParallel)

Thank you!