Training Kit-ML encountered dimension mismatch problem

[lixiang927047]

@ChenFengYe I want to use the framework of MotionGPT to train Kit-ML, but I have encountered a lot of dimensional incompatibility problems, most of which are due to the fact that the data set is only 251 dimensional and the network is set to 263 dimensional (probably because the number of summary joints in the Kit-ML data set is 21). The number of joints for HumanML3D is 22). What do I need to do with the Kit-ML data set before training? Or do different Settings in the training script?

[billl-jiang]

Hello,

For training Kit-ML with MotionGPT and addressing the dimension mismatch issue, you simply need to update the configuration.

MotionGPT/configs/config_h3d_stage1.yaml

Lines 33 to 34 in a56b09f

	DATASET:
	target: mGPT.data.HumanML3D.HumanML3DDataModule

Change mGPT.data.HumanML3D.HumanML3DDataModule to mGPT.data.Kit.KitDataModule in the config file. This should align the dimensions by adapting to the Kit dataset's structure.