Fail to generate the onnx file

[AlyssaZyt]

I need an onnx file to preserve the searching result.
However, there are some errors after I load the pth file to the model.
For example, I loaded the pth file after depth configuration search to the SearchDepthCifarResNet model, Then the system warned me :' Missing key(s) in state_dict and Unexpected key(s) in state_dict: "epoch", "args", "last_checkpoint"."
How to solve the problem? Did I choose the right model? Or how to choose the right model?

[D-X-Y]

Unfortunately, I'm not familiar with ONNX.

[AlyssaZyt]

Unfortunately, I'm not familiar with ONNX.

That's ok. I solved the problem. Thank you.