torch学习 (十五):读取和存储模型引入1 读写Tensor2 读写模型
文章目录引入1 读写Tensor2 读写模型2.1 state_dict2.2 保存和加载模型2.2.1 保存和加载state_dict (推荐)2.2.2 保存整个模型引入 本节介绍如何把在内存中训练好的模型参数进行存储,以及后续的读取[1]^{[1]}[1]。1 读写Tensor torch的save与load函数与numpy的类似:import torchfrom torch import nnif __name__ == '__main__': # Main x