Saved DAGNetwork model is empty when accessed on a differnt machine

4 ビュー (過去 30 日間)
Matthias Kreuzer
Matthias Kreuzer 2022 年 11 月 30 日
I have trained a CNN in pytorch and loaded it into Matlab via ONNX import and saved it is an DAGNetwork. The created DAGNetwork works fine on my computer but when I try to load the same .mat file on another machine, the DAGNetwork cannot be properly loaded. Connections, Layer fields, etc. cannot be accessed. Does anyone know why that is the case?
To my understanding everything that you need to use the model is stored in the DAGNetwork.
  3 件のコメント
Matthias Kreuzer
Matthias Kreuzer 2022 年 12 月 1 日
Yes, the issue was that matlab created a custom flatten layer. I replaced this layer with the Matlab FlattenLayer and then the net could be loaded on another machine.
Sivylla Paraskevopoulou
Sivylla Paraskevopoulou 2022 年 12 月 2 日
In R2022b, MATLAB introduced the Deep Learning Toolbox Converter for PyTorch Models support package. This initial release supports importing image classification models directly from PyTorch. For more details, check out the blog post What’s New in Interoperability with TensorFlow and PyTorch and the importNetworkFromPyTorch documentation page.

サインインしてコメントする。

回答 (0 件)

カテゴリ

Help Center および File ExchangeDeep Learning Toolbox についてさらに検索

製品


リリース

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by