Missing key(s) in state_dict

[zyh4813]

I try to test with optimize_cvae_part.py, but I have the following problem when load the pretrained model "reconstruct.pth" provided in your dataset:

[Kiki0719]

Have u solved it ？