Saving and Loading model pytorch, incremental exercise

 Then let the model was last saved good model training, model loads

# Instantiated model, the optimizer, loss function 
Model MnistModel = (). To (config.device) 
Optimizer = optim.Adam (model.parameters (), LR = 0.01) 

IF os.path.exists ( "Model ./ / mnist_net.pt "): 
    model.load_state_dict (torch.load (" Model ./ / mnist_net.pt ")) 
    optimizer.load_state_dict (torch.load (" Model / mnist_optimizer.pt "))

　　Save model

 

            torch.save(model.state_dict(),"model/mnist_net.pt")
            torch.save(optimizer.state_dict(),"model/mnist_optimizer.pt")