search

OpenMotionLab / MotionGPT

[NeurIPS 2023] MotionGPT: Human Motion as a Foreign Language, a unified motion-language generation model using LLMs
https://motion-gpt.github.io
MIT License
1.38k stars 85 forks source link

VAE - Training #96

Open palmex opened 2 months ago

palmex commented 2 months ago

I'm having a hard time recreating your results. I'm trying to retrain the VAE from scratch: LR 1e-4, Adam, 512 embedding size. The validation error seems to be leveling off, and I don't think brute forcing to 150K epochs would solve this issue.

Would it be possible to share your loss function curves?

Screenshot 2024-05-27 at 8 21 32 AM

palmex commented 3 weeks ago

Need to turn debug mode off