Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Could you share the code used to train t2m_trans_upper body #19

Open
FufenNan opened this issue Aug 10, 2024 · 4 comments
Open

Could you share the code used to train t2m_trans_upper body #19

FufenNan opened this issue Aug 10, 2024 · 4 comments

Comments

@FufenNan
Copy link

Thank you for sharing this outstanding work. I'm wondering how you got the pre-trained trans_upperbody? Could you share the training code?

@exitudio
Copy link
Owner

Thank you for your interest in our work. We haven't cleaned up the upper body training code yet, but if you'd like, you can email me, and I'll send you the uncleaned version of the code.

@FufenNan
Copy link
Author

FufenNan commented Aug 12, 2024

Thank you for your interest in our work. We haven't cleaned up the upper body training code yet, but if you'd like, you can email me, and I'll send you the uncleaned version of the code.

Sure, thanks. I have sent an email.

@jzc6915
Copy link

jzc6915 commented Sep 16, 2024

Thank you for sharing this outstanding work. I'm also interested in the code for the pre-trained trans_upperbody. Could you share the training code?

@exitudio
Copy link
Owner

Hi,
You can get the code here

For training upper-lower:

  1. train VQVAE:
    - see shell script ./experiments/train_vq.sh
    - add "--sep-uplow" option to the train_vq.py
  2. train transformer:
    - see ./experiments/train_trans_uplow.sh

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants