代码之家 › 专栏 › 技术社区 › neo4k

如何保存和使用在PyTorch/TensorFlow/Keras中开发的经过训练的神经网络?

pytorch keras tensorflow

-1

neo4k · 技术社区 · 6 年前

有没有办法在培训和与他人分享模型后保存模型?像楷书?由于网络是一个浮点矩阵的集合,是否可以只提取这些经过训练的权值,并在新数据上运行它来进行预测,而不需要用户安装这些框架呢?我是这些框架的新手,将根据需要作出任何澄清。

1 回复 | 直到 6 年前

randomwalker 6 年前

PyTorch:如中所述 this post 这两个过程都要求用户至少安装一个张量计算框架,例如用于有效的矩阵乘法。

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

3 年前

Chris Ranck · Keras时间序列生成器-输入为“X”长,输出为“X-长”

3 年前

DPM · 将自定义掩码传递给LSTM数据以进行培训和验证

3 年前

Luciana Carota · 为什么Keras的R“预测”函数不能识别正确的输入维度?

3 年前

Sarah Elnaggar · 我使用Gensim Doc2vec进行图形嵌入,然后在keras中使用两层深度神经网络进行二元分类

3 年前

user836026 · 在pytorch中复制权重时语法无效

3 年前

mad · 如何从一个定制的神经网络模型中获得逻辑和概率

3 年前

Archit · 无法使用自定义距离函数加载带有lambda函数的keras模型

6 年前