Fail to generate the onnx file
AlyssaZyt opened this issue · comments
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?
Unfortunately, I'm not familiar with ONNX.
Unfortunately, I'm not familiar with ONNX.
That's ok. I solved the problem. Thank you.