search

D-X-Y / AutoDL-Projects

Automated deep learning algorithms implemented in PyTorch.
MIT License
1.57k stars 282 forks source link

Fail to generate the onnx file #114

Open AlyssaZyt opened 2 years ago

AlyssaZyt commented 2 years ago

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 commented 2 years ago

Unfortunately, I'm not familiar with ONNX.

AlyssaZyt commented 2 years ago

Unfortunately, I'm not familiar with ONNX.

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