PyTorch深度学习实践 保存提取 def save_net(): #保存 torch.save(模型名称, 'net.pkl') #保存整个神经网络 #或者只保留节点 参数 torch.save(模型名称.state_dict(), 'net_params.pkl') def restore_net(): #提取 net