摘要: 1.三个核心函数 介绍一系列关于 PyTorch 模型保存与加载的应用场景,主要包括三个核心函数: (1)torch.save 其中,应用了 Python 的 pickle 包,进行序列化,可适用于模型Models,张量Tensors,以及各种类型的字典对象的序列化保存. (2)torch.load 阅读全文
posted @ 2020-08-19 18:05 雨后观山色 阅读(5190) 评论(0) 推荐(0)