保存和加载自动编码器

Question

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

AE 是使用 torch.nn.Module 包创建的，有 4 个 layers.Unlike ALS 用于 AE 保存和加载模型我不是 missing.Can 谁能帮我找到保存和加载模型的方法加载自动编码器。

Answer 1

torch模块提供save和load方法

它将模型保存为 .pth 文件

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

还有加载方法

torch.load(.pth_file)

保存和加载自动编码器

Saving and loading an autoencoder

python

prediction

neural-network

autoencoder

pyspark