Yes, you are correct. To visualise a Pytorch model with Netron, you have to save it in ONNX as it also saves the graph. I use torch.onnx.export() for this. With torch.save() you can just visualise individual blocks in Netron and see the weights, but not the network graph.