speechbrain.utils.pretrained module

预训练模型的训练工具

作者 * Artem Ploujnikov 2021

摘要

函数

save_for_pretrained

从找到的最佳检查点保存预训练模型所需的必要文件。

参考

speechbrain.utils.pretrained.save_for_pretrained(hparams, min_key=None, max_key=None, ckpt_predicate=None, pretrainer_key='pretrainer', checkpointer_key='checkpointer')[source]

从找到的最佳检查点保存预训练模型所需的必要文件。此函数的目的是为 Pretrainer 导出模型

参数:

  • hparams (dict) – 超参数文件

  • min_key (str) – 用于查找最佳检查点的键（值越低越好）。默认情况下，传递给 self.checkpointer.recover_if_possible()。

  • max_key (str) – 用于查找最佳检查点的键（值越高越好）。默认情况下，传递给 self.checkpointer.recover_if_possible()。

  • ckpt_predicate (callable) – 用于定位检查点的过滤谓词

  • pretrainer_key (str) – 存储 pretrainer 的键

  • checkpointer_key (str) – 存储 checkpointer 的键

返回:

saved – 保存是否成功

返回类型:

bool