Pytorch 保存 tensor

General Purpose image with conda installed to be used in PyTorch CI/CD

Official images for TensorFlow Serving (http://www.tensorflow.org/serving)

PyTorch Lightning image

1. 模型存储接口 首先我们知道不论是保存模型还是参数都需要用到torch.save()。 对于torch.save()有两种保存方式： a. 只保存神经网络的训练模型的参数，save的对象是model.state_dict()； b. 既保存整个神经网络的的模型结构又保存模型参数，那么save