Update seeds for single-GPU reproducibility (#3789) · 92d49fde35 - yolov5 - 拓恒技术版本管理系统

Update seeds for single-GPU reproducibility (#3789)

For seed=0 on single-GPU.

modifyDataloader

Glenn Jocher

GitHub 3 years ago

parent

8e7f285051

commit

92d49fde35

No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23

1 changed files with 1 additions and 1 deletions

Split View

							
								+ 1
								
								- 1
							
train.pyView File
								
			@@ -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