How to transfer a trained model from Matlab to Pytorch?

[Jimmy_Xiaoke_Shen]

Any ideas about how to transfer a trained model from Matlab to Pytorch?

Here is the discussion about how to transfer from Pytorch to Matlab. However, I need the inverse thing…

How to transfer a trained model from Pytorch to Matlab?

Hi, Everyone. I have a trained CNN model in Pytorch and I want to transfer it to the Matlab. is there any solution for this purpose? Thank you for advising me on this topic. Reza.

Thanks in advance for any suggestions.

[ptrblck]

You could try to save the MATLAB model parameters as .mat files.
After re-building the model in PyTorch, you could load these parameters using scipy.io and assign them to your PyTorch model.
Make sure you check the shaped and also possible flipped kernels, since some frameworks flip the conv kernels to use a convolution instead of the correlation.

[Jimmy_Xiaoke_Shen]

Thanks for the suggestion. I’ll try. By the way, for the pytorch, do we use the convolution or correlation? It seems that we are using the correlation, right?

[ptrblck]

Yep, is the correlation, since the kernels are not flipped again.

[Jimmy_Xiaoke_Shen]

Cool. It is much clear now. Thanks a lot.

[Jibanul]

@Jimmy_Xiaoke_Shen, I’m trying to do the same thing. Did you able to do this?

[Jimmy_Xiaoke_Shen]

@ptrblck

ing the model in PyTorch, you could load these parameters using scipy.io and assign them to your PyTorch model.
Make sure you check the shaped and also possible flipped kernels, since some frameworks flip the conv kernels to use a convolution in

Sorry for the late reply as I haven’t login for a long time. I gave up using the Matlab weights so I don’t know whether there is a good way to do that.