fastNLP.io.model_io

用于载入和保存模型

class fastNLP.io.model_io.ModelLoader[源代码]

别名 fastNLP.io.ModelLoader fastNLP.io.model_io.ModelLoader

用于读取模型

static load_pytorch(empty_model, model_path)[源代码]

从 ".pkl" 文件读取 PyTorch 模型

参数:
  • empty_model -- 初始化参数的 PyTorch 模型
  • model_path (str) -- 模型保存的路径
static load_pytorch_model(model_path)[源代码]

读取整个模型

参数:model_path (str) -- 模型保存的路径
class fastNLP.io.model_io.ModelSaver(save_path)[源代码]

别名 fastNLP.io.ModelSaver fastNLP.io.model_io.ModelSaver

用于保存模型

Example:

saver = ModelSaver("./save/model_ckpt_100.pkl")
saver.save_pytorch(model)
__init__(save_path)[源代码]
参数:save_path -- 模型保存的路径
save_pytorch(model, param_only=True)[源代码]

把 PyTorch 模型存入 ".pkl" 文件

参数:
  • model -- PyTorch 模型
  • param_only (bool) -- 是否只保存模型的参数（否则保存整个模型）