PyTorch weights support

[ardeal]

Hi,

I use Netron tool to visualize the pth file from the following link:
https://download.pytorch.org/models/vgg19_bn-c79401a0.pth

If I use Netron tool to visualize some other format file, there are arrow-lines to connect each layer of the network.

but the visualized output network of pth file seems not correct. there is no arrow-lines to connect each layer....

Thanks,
Ardeal

[ardeal]

@lutzroeder ,
Hi,

I am still confused about how to convert the pth file to onnx file.

at present, I do save the pth file with code:
torch.save(model.state_dict(), filepath)

according to #133 , I got 2 understandings:

1. re-train the model and save the pth file with code: torch.save(model, filepath)
2. don't need to re-train the model. I need to use some code to convert the pth file to onnx file. what the code should be? is there any example code?

which of my upper understandings is correct?

Thanks,
Ardeal

[lutzroeder]

wget https://api.ngc.nvidia.com/v2/models/nvidia/megatron_bert_345m/versions/v0.1_cased/zip -O megatron_bert_345m_v0.1_cased.zip

[lutzroeder]

Duplicate of #720