将火炬模型(torch.save)转换为可以用基本 Python 处理的矩阵公式

Posted

技术标签:

【中文标题】将火炬模型(torch.save)转换为可以用基本 Python 处理的矩阵公式【英文标题】:Convert torch model (torch.save) into matricial formulas that can be handled with basic Python 【发布时间】:2021-07-25 05:46:23 【问题描述】:

请您告诉我将 Torch 模型 (torch.save) 转换为可以使用 numpy 或基本 Python 操作的代数矩阵/方程是否可行,而无需安装 Torch 和其他相关库(占用很多空间)?在肯定的情况下,您能否给我一些提示或解释的链接?非常感谢。

【问题讨论】:

【参考方案1】:

如果没有您自己的大量工作,我不知道有什么方法可以做到这一点。基本上,您必须将大部分 pytorch 库移植到 numpy,这将是一个巨大的项目。如果空间是一个问题,请检查您是否可以通过例如使用较早的 Torch 版本或仅使用 pytorch 的 CPU 版本来节省一些空间。

【讨论】:

以上是关于将火炬模型(torch.save)转换为可以用基本 Python 处理的矩阵公式的主要内容,如果未能解决你的问题,请参考以下文章

pytorch保存模型遇到点问题

Torch.load()使用方式

pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件

如何将 matplotlib 频谱图图像转换为火炬张量

保存和加载模型

PyTorch保存模型断点以及加载断点继续训练