torch Save Load Model

  • Save Model

    torch.save(my_model.state_dict(), "params.pkl")
  • Load Model

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

Guess you like

Origin www.cnblogs.com/rise0111/p/11621640.html