ChenFengYe/motion-latent-diffusion

Training configuration on KIT dataset

Dean-UQ opened this issue · 8 comments

Hi,
I would like to train MLD on KIT dataset. What should I tune in this configuration file https://github.com/ChenFengYe/motion-latent-diffusion/blob/main/configs/config_vae_humanml3d.yaml ? Its JOINT_TYPE may be configure as 'mmm'?
Many Thanks

If you use the kit from humanml3d (like below), you only need to change
humanml3d => kit


DATASETS: ['humanml3d'] # Evaluating datasets

DATASETS: ['humanml3d'] # training datasets

Importantly, you do not need to change the joint type.
JOINT_TYPE: 'humanml3d' # join type

You can also check #2

We do not suggest you download KIT from its official website, you can check the KIT provided by HumanML3D.
The below is from the readme of HumanML3D

For KIT-ML dataset, you could directly download [Here]. Due to the distribution policy of AMASS dataset, we are not allowed to distribute the data directly. We provide a series of script that could reproduce our HumanML3D dataset from AMASS dataset.

https://drive.google.com/drive/folders/1MnixfyGfujSP-4t8w_2QvjtTVpEKr97t

are there any more config changes that need to be applied to reproduce the results reported in table 2?

weleen commented

@nefeliandreou have you reproduced the reported results?

@weleen Do you need the KIT pre-trained model or the implementation of this training process?

weleen commented

@weleen Do you need the KIT pre-trained model or the implementation of this training process?

Thank you for your prompt response, it’s really nice if you could provide the pretrained model and log files.

@weleen Do you need the KIT pre-trained model or the implementation of this training process?

Hello, I've encountered issues while training KIT. Could you provide the pre-trained model? Thank you very much for your contributions to the research community.

Hi @ChenFengYe, could you please provide the checkpoint of MLD pre-trained on KIT? I would need it as well.
Thank you very much in advance.

Hi @ChenFengYe, I've encountered same issues while training KIT. Could you provide the pre-trained model? Thank you very much.