The checkpoint pretrained on KIT-ML dataset

[qinghuannn]

Hi. Thanks for your nice work!
Could you release the checkpoint pretrained on KIT-ML dataset?

[vinit-katariya]

@exitudio thank you for your great work! Would it be possible for you to release the checkpoint pre-trained on the KIT-ML dataset?

[exitudio]

Hi,
You can download the pretrain KIT. I haven't tested it, so there may be some bugs in KIT. You can investigate the saved source code in ./source

P.S. For evaluation, the masking strategy is different from HumanML3D—it keeps the highest probability. Replacing ./models/mask_transformer/transformer.py with ./source/models/mask_transformer/transformer__8_res_2iter_noRepredAll_full.py should slightly improve the score.