fastNLP.io.model_io module

用于载入和保存模型

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

基类:object

别名 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)[源代码]

基类:object

别名 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) – 是否只保存模型的参数(否则保存整个模型)