Sampling process running error

[zhuiguangzhe-zz]

I can train normally, but I encounter a key value missing problem when sampling. What is the reason?

[isaacjones99]

The reason for this is that the model you are loading the state dict into is different than the model you have defined in your code. This could be because the parameters you're using for sampling are different than the ones you used for training. Or, maybe you're trying to load the wrong model file.