neuralpredictor.pytorch icon indicating copy to clipboard operation
neuralpredictor.pytorch copied to clipboard

Saved Models?

Open GrayGriffin opened this issue 4 years ago • 1 comments

Is there a way to view the models saved after the training? I can't open the .pyc files even though they have the Python logo.

GrayGriffin avatar Oct 01 '21 10:10 GrayGriffin

I think you need to call torch.save(model.state_dict() to save the model and call torch.load to reload the model. See tutorial here: https://pytorch.org/tutorials/beginner/saving_loading_models.html

pyc file are "python cache" files. They are not checkpoints.

ultmaster avatar Oct 02 '21 02:10 ultmaster