Save Model
torch.save(my_model.state_dict(), "params.pkl")
Load Model
先初始化model网络结构 model.load_state_dict(torch.load("params.pkl"))
torch Save Load Model
Guess you like
Origin www.cnblogs.com/rise0111/p/11621640.html
Recommended
Ranking