speechbrain.utils.seed 模块

用于可复现性的种子工具。

作者

Adel Moumen 2024

摘要

函数

seed_everything

在 torch、numpy 和 Python 的 random 模块中设置伪随机数生成器种子的函数。

参考

speechbrain.utils.seed.seed_everything(seed: int = 0, verbose: bool = True, deterministic: bool = False) → int[源代码]

在 torch、numpy 和 Python 的 random 模块中设置伪随机数生成器种子的函数。关于 DDP 的重要说明：所有 DDP 进程具有相同的种子。这对于确保 require_grad 未设置为 True 的参数在所有进程中相同非常重要。如果想要构建自定义数据采样器，必须考虑到这一点，因为进程将选择相同的样本……SpeechBrain 会在内部处理好这一点。

参数:

seed (int) – 全局随机状态的整数种子值。
verbose (bool) – 是否在每个 rank 上打印一条消息，说明正在设置的种子。
deterministic (bool) – 是否为确定性操作设置种子。

返回值:

已设置的种子。

返回类型:

int