From 92d49fde354126405b866c69b055fb9156115d55 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 26 Jun 2021 15:42:40 +0200 Subject: [PATCH] Update seeds for single-GPU reproducibility (#3789) For seed=0 on single-GPU. --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3390e83..9ac12b1 100644 --- a/train.py +++ b/train.py @@ -84,7 +84,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary # Configure plots = not evolve # create plots cuda = device.type != 'cpu' - init_seeds(2 + RANK) + init_seeds(1 + RANK) with open(data) as f: data_dict = yaml.safe_load(f) # data dict