Loading .pt file into matlab

53 visualizzazioni (ultimi 30 giorni)
Bao Ngo
Bao Ngo il 19 Apr 2022
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.

Risposte (1)

Sivylla Paraskevopoulou
Sivylla Paraskevopoulou il 7 Ott 2022
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.

Categorie

Scopri di più su Image Data Workflows in Help Center e File Exchange

Prodotti


Release

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by