BAMM icon indicating copy to clipboard operation
BAMM copied to clipboard
verified publisher icon exitudio

The checkpoint pretrained on KIT-ML dataset

Open qinghuannn opened this issue 11 months ago • 2 comments

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

qinghuannn avatar Jan 08 '25 09:01 qinghuannn

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

vinit-katariya avatar Mar 07 '25 19:03 vinit-katariya

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.

exitudio avatar Mar 11 '25 15:03 exitudio