保存和加载自动编码器

Question

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

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

Answer 1

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

它将模型保存为.pth文件

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

它也有加载方式

torch.load(.pth_file)