HumanML3D
HumanML3D copied to clipboard
Vectorizing `amass_to_pose` in `raw_pose_processing`
Hello,
I think amass_to_pose can be vectorized (not iterating a single pose at a time but all poses in a sequence (a npz file) at a single time.).
I tried this and it gave about >10x boost in speed. (since my graphics card GTX 1660 Ti gave me a CUDA memory allocation error, I calculated in cpu though.).
If it's not a resolved issue, I may issue a PR.
Thanks,
Young
Hi I am facing a similar issue while processing (amass_to_pose) KIT dataset:
Can you please help out
My GPU is 2080 Ti