torch.save(net1, 'net.pkl') #保存entire net整个网络
torch.save(net1.state_dict(), 'net_params.pkl') #保存参数```
pytorch的学习
最新推荐文章于 2022-06-14 17:28:30 发布
torch.save(net1, 'net.pkl') #保存entire net整个网络
torch.save(net1.state_dict(), 'net_params.pkl') #保存参数```