facebookresearch / AGRoL

Code release for "Avatars Grow Legs Generating Smooth Human Motion from Sparse Tracking Inputs with Diffusion Model", CVPR 2023
Other
238 stars 24 forks source link

Query Regarding Training Duration and Multi-GPU Support #11

Closed ljqnb closed 1 year ago

ljqnb commented 1 year ago

Hi, could you please tell me about the total training time and total steps you used for your model? I am using the default parameter in your code which sets numsteps = 6000000, but the training on eight A100 GPUs takes nearly 10 days. Could it be possible that this code does not support multi-GPU usage?