CODEWITHSUNDEEP
pythonneural-networkpysparkpredictionautoencoder
ANANDHU AJAYAKUMAR
ANANDHU AJAYAKUMAR

Reputation: 55

Saving and loading an autoencoder

I am trying to make a predictor using Auto Encoder (AE) and Alternating Least Squares (ALS) methods. ALS is created using pyspark.mllib.recommendation package. I am able to save the ALS model and reuse it by model.save() and pyspark.mllib.recommendation.MatrixFactorizationModel.load() methods.

AE is created using torch.nn.Module package and has 4 layers.Unlike ALS for AE saving and loading model is not something I am missing.Can anyone help me to find a way to save and load an auto encoder.

Upvotes: 2

Views: 936

Answers (1)

Anandhu Ajayakumar
Anandhu Ajayakumar

Reputation: 136

The torch module provides save and load methods.

It saves the model as .pth files 

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

It also has load method

torch.load(.pth_file)

Upvotes: 1

Related Questions