MMM icon indicating copy to clipboard operation
MMM copied to clipboard

Could you share the code used to train t2m_trans_upper body

Open FufenNan opened this issue 1 year ago • 4 comments

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

FufenNan avatar Aug 10 '24 23:08 FufenNan

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.

exitudio avatar Aug 11 '24 20:08 exitudio

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.

FufenNan avatar Aug 12 '24 02:08 FufenNan

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?

jzc6915 avatar Sep 16 '24 08:09 jzc6915

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

exitudio avatar Sep 16 '24 16:09 exitudio