set_seed¶

torchtune.training.set_seed(seed: Optional[int] = None, debug_mode: Optional[Union[int, str]] = None) → int[source]¶

设置常用库中伪随机数生成器种子的函数。

这将为PyTorch、NumPy和python.random模块设置种子。对于分布式作业，每个本地进程都会设置自己的种子，计算种子 + 排名。有关更多详细信息，请参阅https://pytorch.org/docs/stable/notes/randomness.html。

Parameters:

seed (可选[int]) – 整数值种子。如果为 None，将生成并设置一个随机种子。
debug_mode (可选[联合[str, int]]) –
控制PyTorch中确定性操作的debug_mode设置。
- 如果None，不设置任何PyTorch全局值。
- 如果“default”或0，不对非确定性操作发出错误或警告，并启用PyTorch CuDNN基准测试。
- 如果“warn”或1，对非确定性操作发出警告并禁用PyTorch CuDNN基准测试。
- 如果“error”或2，对非确定性操作发出错误并禁用PyTorch CuDNN基准测试。
- 更多详情，请参见torch.set_deterministic_debug_mode()和 https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms。

Returns:

当前种子

Return type:

int

Raises:

ValueError – 如果输入的种子值超出所需范围。