how to use the trained model ckpt for test?
ShungJhon opened this issue · 2 comments
ShungJhon commented
I've followed your instruction to train the mld model. And I got the checkpoint after 2000epochs. Now I want to use this ckpt for test just like the 1222_mld_humanml3d_FID041.ckpt in your template. But the state_dict is wrong.
RuntimeError: Error(s) in loading state_dict for MLD:
Missing key(s) in state_dict: "denoiser.decoder.layers.0.self_attn.in_proj_weight", "denoiser.decoder.layers.0.self_attn.in_proj_bias", ………………
Is there any additional step to get the right ckpt? or any additional setup in the config file?
您好，我想用我训练好的模型测试性能，但是load 训练的checkpoint遇到了问题，请问我需要添加其他额外的步骤吗？
ShungJhon commented
请问是不是因为训练mld的时候是把vae也存到了ckpt里
ChenFengYe commented
是的 ckpt里有VAE参数