保存并加载自动编码器

Question

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

AE是使用torch.nn.Module包创建的，有4层。与ALS不同，AE保存和加载模型不是我缺少的东西。任何人都可以帮我找到保存和加载自动编码器的方法。 / p>

Answer 1

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

将模型保存为.pth文件

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

它还有加载方法

torch.load(.pth_file)