自動エンコーダーの保存と読み込み

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') 

また、loadメソッドを持ってい

torch.load(.pth_file)