[pytroch] 存储torch到文件

Posted 2021-05-17 schaffen

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[pytroch] 存储torch到文件相关的知识，希望对你有一定的参考价值。

# 存储数据
import scipy.io as sio
output_file = "xx.mat"
output_dict = {"a":data,"b":data}
sio.savemat(output_file,output_dict)

# 读取数据
sio.loadmat(output_file)

以上是关于[pytroch] 存储torch到文件的主要内容，如果未能解决你的问题，请参考以下文章