最近每次保存和载入模型都会忘记代码怎么写,写在这里记住。
保存模型参数
torch.save(net.state_dict(), Path)
net.state_dicct()是一个字典,net中层的名称和参数的对应。Path为要保存的模型的路径和名称。
加载模型参数
net.load_state_dict(torch.load(Path))
Path是要加载的保存好的模型参数的路径
最近每次保存和载入模型都会忘记代码怎么写,写在这里记住。
保存模型参数
torch.save(net.state_dict(), Path)
net.state_dicct()是一个字典,net中层的名称和参数的对应。Path为要保存的模型的路径和名称。
加载模型参数
net.load_state_dict(torch.load(Path))
Path是要加载的保存好的模型参数的路径