📌  相关文章
📜  pytorch 保存模型 - Python 代码示例

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

代码示例4
torch.save(model.state_dict(), PATH)