Hi. thanks for the repository.

I'm failing to visualize a PyTorch model that I save with the following line:

torch.save(model.state_dict(), 'model.pth')

and then I get this Error from Netron :

"Error loading PyTorch model. File does not contain a model graph. Use 'torch.save()' to save both the graph and tensor data."

But when I convert it to ONNX in the following way, I'm able to visualize the ONNX model :
torch.onnx.export(model, img, 'model.onnx')

Also when I tried to save the full model in PyTorch (without the ".state_dict()" part), I don't get the error but it doesn't show the correct Graph and only a single-node graph basically is displayed.

Any clue how to resolve this issue?