📌  相关文章
📜  火炬保存状态字典 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:14.204000             🧑  作者: Mango

代码示例1
torch.save(model.state_dict(), PATH)
model.load_state_dict(torch.load(PATH))