Loading .pt file into matlab

44 visualizaciones (últimos 30 días)
Bao Ngo
Bao Ngo el 19 de Abr. de 2022
Respondida: Sivylla Paraskevopoulou el 7 de Oct. de 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.

Respuestas (1)

Sivylla Paraskevopoulou
Sivylla Paraskevopoulou el 7 de Oct. de 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.

Categorías

Más información sobre Image Data Workflows en Help Center y File Exchange.

Productos


Versión

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by