第二十篇—读取和存储

部分一：读写tensor

torch.save：存储；

torch.load：读取；

代码实现：

结果：

部分二：模型的state_dict

代码实现：

结果：

注意：

只有可学习参数的层（卷积层、线性层等）才有state_dict；优化器也有state_dict，包含关于优化器状态以及所使用的超参数信息，如：

结果：

部分三：保存和加载模型

方法一：保存和加载模型的state_dict：

结果：

方法二：保存和加载整个模型

结果：

保存和加载整个模型，没有模型构建的过程，即不需要再重新实例化模型了。

保存state_dict()和保存整个模型的文件大小对比：