hongsukchoi / TCMR_RELEASE

Official Pytorch implementation of "Beyond Static Features for Temporally Consistent 3D Human Pose and Shape from a Video", CVPR 2021
MIT License
274 stars 39 forks source link

Two graphics cards? #13

Open Changboyang opened 2 years ago

Changboyang commented 2 years ago

Hi, Do you know how this project can be run on two graphics cards ?

hongsukchoi commented 2 years ago

Yes, you can. Refer to this for multi-GPU training of RNNs https://discuss.pytorch.org/t/why-do-we-need-flatten-parameters-when-using-rnn-with-dataparallel/46506