序列化的相关语义
译者:yuange250
最佳方案
保存模型的推荐方法
Pytorch主要有两种方法可用于序列化和保存一个模型。
第一种只存取模型的参数(更为推荐): 保存参数:
读取参数:
第二种方法则将整个模型都保存下来:
读取的时候也是读取整个模型:
在第二种方法中, 由于特定的序列化的数据与其特定的类别(class)相绑定,并且在序列化的时候使用了固定的目录结构,所以在很多情况下,如在其他的一些项目中使用,或者代码进行了较大的重构的时候,很容易出现问题。
译者:yuange250
Pytorch主要有两种方法可用于序列化和保存一个模型。
第一种只存取模型的参数(更为推荐): 保存参数:
读取参数:
第二种方法则将整个模型都保存下来:
读取的时候也是读取整个模型:
在第二种方法中, 由于特定的序列化的数据与其特定的类别(class)相绑定,并且在序列化的时候使用了固定的目录结构,所以在很多情况下,如在其他的一些项目中使用,或者代码进行了较大的重构的时候,很容易出现问题。