Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 1.04 KB

set_random_seed.md

File metadata and controls

20 lines (14 loc) · 1.04 KB

设置随机种子

如果想要在训练时指定随机种子,可以使用以下命令:

python ./tools/train.py \
    ${CONFIG} \                               # 配置文件路径
    --cfg-options randomness.seed=2023 \      # 设置随机种子为 2023
    [randomness.diff_rank_seed=True] \        # 根据 rank 来设置不同的种子。
    [randomness.deterministic=True]           # 把 cuDNN 后端确定性选项设置为 True
# [] 代表可选参数,实际输入命令行时,不用输入 []

randomness 有三个参数可设置,具体含义如下:

  • randomness.seed=2023 ,设置随机种子为 2023。

  • randomness.diff_rank_seed=True,根据 rank 来设置不同的种子,diff_rank_seed 默认为 False。

  • randomness.deterministic=True,把 cuDNN 后端确定性选项设置为 True,即把torch.backends.cudnn.deterministic 设为 True,把 torch.backends.cudnn.benchmark 设为False。deterministic 默认为 False。更多细节见 https://pytorch.org/docs/stable/notes/randomness.html。