【深度之眼】Pytorch框架班第五期-Week7【任務1】第一節：模型保存與加載

序列化與反序列化 序列化與反序列化表示的是內存與硬盤之間的序列關係。 1、Pytorch中的序列化與反序列化 1、torch.save 主要參數： obj: 對象 f: 輸出路徑 2、torch.load 主要參數： f: 文件路徑 map_location: 指定存放位置，cpu or gpu Pytorch中保存模型的兩種方法 法1：保存整個Module torch.save(net, pat

>>阅读原文<<