保存和加载自动编码器答案

【问题标题】：Saving and loading an autoencoder保存和加载自动编码器
【发布时间】：2018-04-04 20:06:41
【问题描述】：

我正在尝试使用 自动编码器 (AE) 和 交替最小二乘法 (ALS) 方法进行预测。 ALS 是使用 pyspark.mllib.recommendation 包创建的。我能够保存 ALS 模型并通过 model.save() 和 pyspark.mllib.recommendation.MatrixFactorizationModel.load() 方法重用它。

AE 是使用 torch.nn.Module 包创建的，有 4 层。与 ALS 不同，用于 AE 保存和加载模型不是我缺少的东西。谁能帮我找到一种方法来保存和加载自动编码器。

【问题讨论】：

标签： python neural-network pyspark prediction autoencoder

【解决方案1】：

torch 模块提供保存和加载方法。

将模型保存为 .pth 文件

torch.save(model,'model.pth')

它也有加载方法

torch.load(.pth_file)

【讨论】：