- 保存模型
torch.save(my_model.state_dict(), "params.pkl")
- 加载模型
先初始化model网络结构 model.load_state_dict(torch.load("params.pkl"))
原文地址:https://www.cnblogs.com/rise0111/p/11621640.html
时间: 2024-10-09 15:28:31
torch.save(my_model.state_dict(), "params.pkl")
先初始化model网络结构
model.load_state_dict(torch.load("params.pkl"))
原文地址:https://www.cnblogs.com/rise0111/p/11621640.html