标签:code 保存加载 model dict ORC load 加载 ram stat
保存模型
torch.save(my_model.state_dict(), "params.pkl")
加载模型
先初始化model网络结构
model.load_state_dict(torch.load("params.pkl"))
标签:code 保存加载 model dict ORC load 加载 ram stat
原文地址:https://www.cnblogs.com/rise0111/p/11621640.html