Pytorch保存和载入模型参数

最近每次保存和载入模型都会忘记代码怎么写,写在这里记住。

保存模型参数

torch.save(net.state_dict(), Path)

net.state_dicct()是一个字典,net中层的名称和参数的对应。Path为要保存的模型的路径和名称。

加载模型参数

net.load_state_dict(torch.load(Path))

Path是要加载的保存好的模型参数的路径

你可能感兴趣的:(PyTorch)