speechbrain.utils.pretrained 模块

预训练模型的训练工具

作者 * 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]

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

Parameters:

  • hparams (dict) – 超参数文件

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

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

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

  • pretrainer_key (str) – 预训练器存储的键

  • checkpointer_key (str) – 存储检查点的键

Returns:

saved – 保存是否成功

Return type:

bool