部分一：讀寫tensor

torch.save：存儲；

torch.load：讀??；

代碼實(shí)現(xiàn)：

結(jié)果：

部分二：模型的state_dict

代碼實(shí)現(xiàn)：

結(jié)果：

注意：

只有可學(xué)習(xí)參數(shù)的層（卷積層、線性層等）才有state_dict；優(yōu)化器也有state_dict，包含關(guān)于優(yōu)化器狀態(tài)以及所使用的超參數(shù)信息，如：

結(jié)果：

部分三：保存和加載模型

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

結(jié)果：

方法二：保存和加載整個(gè)模型

結(jié)果：

保存和加載整個(gè)模型，沒有模型構(gòu)建的過程，即不需要再重新實(shí)例化模型了。

保存state_dict()和保存整個(gè)模型的文件大小對比：