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

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

pytorch keras tensorflow

-1

neo4k · 技术社区 · 7 年前

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

1 回复 | 直到 7 年前

randomwalker 7 年前

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

推荐文章

Saffy · 如何在IterableDataset上应用最小最大缩放?

1 年前

sanjeev mk · 通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

2 年前

Anonymous · 如何为零维火炬张量赋值?

2 年前

JohnnyWang97 · getattr引起的有趣错误

2 年前

Kamugg · 在PyTorch中使用不同分辨率图像训练DeepLabV3的最佳实践

2 年前

Stocavista · 无法在python中将float 64转换为float 32

2 年前

efwefwefwefwefw wefwefwefwef · 如何在PyTorch Conv1d层中仅在一侧应用填充?

2 年前

Okhr · 运行时错误:CUDA错误:在带有GTX 1660 Super的Debian 12虚拟机上不支持此操作

2 年前

Fatemeh · 如何从使用nn训练和保存的模型加载检查点。DataParallel到不使用nn的模型上。DataParallel?

2 年前

Twenkid · 将GPT2 h5型号转换为割炬,以转换为ggml形状不匹配

2 年前