Loading .pt file into matlab

47 次查看(过去 30 天)
Bao Ngo
Bao Ngo 2022-4-19
Hi I have a .pt file containing a list of tensors like the picture below. I want to ask how can I load these tensors into Matlab? (P/s: I get these tensors after performing VGG11 on Pytorch) I am also trying to save it as an onnx file but when I load it to Matlab, I got an error.

回答(1 个)

Sivylla Paraskevopoulou
In R2022b we introduced the Deep Learning Toolbox Converter for PyTorch Models support package. This initial release supports importing image classification models. Support for other model types will be added in future releases. Use the function importNetworkFromPyTorch to import a PyTorch model. Make sure that the PyTorch model that you are importing is pretrained and traced.
For more details, check out the blog post What’s New in Interoperability with TensorFlow and PyTorch and the importNetworkFromPyTorch documentation page.

类别

Help CenterFile Exchange 中查找有关 Call Python from MATLAB 的更多信息

产品


版本

R2022a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by