HumanML3D icon indicating copy to clipboard operation
HumanML3D copied to clipboard

Vectorizing `amass_to_pose` in `raw_pose_processing`

Open jojo978 opened this issue 2 years ago • 2 comments

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

jojo978 avatar Nov 20 '23 18:11 jojo978

Hi I am facing a similar issue while processing (amass_to_pose) KIT dataset:

Screenshot 2024-08-13 094350

Can you please help out

sridharkamath avatar Aug 13 '24 04:08 sridharkamath

My GPU is 2080 Ti

sridharkamath avatar Aug 13 '24 04:08 sridharkamath