search

jia-zhuang / pytorch-multi-gpu-training

整理 pytorch 单机多 GPU 训练方法与原理
706 stars 80 forks source link

ddp_train.py 里没有固定随机种子操作 #1

Closed xubin1991 closed 2 years ago

xubin1991 commented 2 years ago

Readme 里面的ddp 代码里有以下行:

# 固定随机种子
seed = 42
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed)

但是在代码ddp_train.py里没有这些行, 这是故意的吗?

jia-zhuang commented 2 years ago

要加的，代码里忘记了，我加上