torch.save(net1, 'net.pkl') #保存entire net整个网络 torch.save(net1.state_dict(), 'net_params.pkl') #保存参数```