pytorch 中模型的保存與加載，增量訓練

 讓模型接着上次保存好的模型訓練，模型加載

#實例化模型、優化器、損失函數
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"))

　　模型保存

 

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