@ptrblck ，
Thank you！
if the pth file contains only state_dict, is there a way to export the model to onnx?
If the pth file contains the model, I should export the model to onnx like this:
model = torch.load(PATH)
model.eval()
torch.onnx.export(model, dummy_input, "alexnet.onnx", verbose=True, input_names=input_names, output_names=output_names)