torch保存加载模型

• 保存模型

  torch.save(my_model.state_dict(), "params.pkl")

• 加载模型

  先初始化model网络结构
  model.load_state_dict(torch.load("params.pkl"))