Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

Set a seed of generator with an option for more randomness when training several models with different seeds (#10486)

Browse Source 
* set seed with parameter

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>

* make seed to be a large number

* set seed with a parameter

* set a seed of dataloader with opt for more randomness

Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Yonghye KwonandGlenn Jocher authored and GitHub committed 2022-12-19 10:27:34 +01:00
1 parent b2a0f1cdc5
commit 10e93d295f
3 files changed
+8 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

train.py
+2 -1
View File
@@ -198,7 +198,8 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
image_weights=opt.image_weights, image_weights=opt.image_weights,
quad=opt.quad, quad=opt.quad,
prefix=colorstr('train: '), prefix=colorstr('train: '),
shuffle=True) shuffle=True,
seed=opt.seed)
labels = np.concatenate(dataset.labels, 0) labels = np.concatenate(dataset.labels, 0)
mlc = int(labels[:, 0].max()) # max label class mlc = int(labels[:, 0].max()) # max label class
assert mlc < nc, f'Label class {mlc} exceeds nc={nc} in {data}. Possible class labels are 0-{nc - 1}' assert mlc < nc, f'Label class {mlc} exceeds nc={nc} in {data}. Possible class labels are 0-{nc - 1}'
utils/dataloaders.py
+3 -2
View File
@@ -115,7 +115,8 @@ def create_dataloader(path,
image_weights=False, image_weights=False,
quad=False, quad=False,
prefix='', prefix='',
shuffle=False): shuffle=False,
seed=0):
if rect and shuffle: if rect and shuffle:
LOGGER.warning('WARNING ⚠️ --rect is incompatible with DataLoader shuffle, setting shuffle=False') LOGGER.warning('WARNING ⚠️ --rect is incompatible with DataLoader shuffle, setting shuffle=False')
shuffle = False shuffle = False
@@ -140,7 +141,7 @@ def create_dataloader(path,
sampler = None if rank == -1 else distributed.DistributedSampler(dataset, shuffle=shuffle) sampler = None if rank == -1 else distributed.DistributedSampler(dataset, shuffle=shuffle)
loader = DataLoader if image_weights else InfiniteDataLoader # only DataLoader allows for attribute updates loader = DataLoader if image_weights else InfiniteDataLoader # only DataLoader allows for attribute updates
generator = torch.Generator() generator = torch.Generator()
generator.manual_seed(6148914691236517205 + RANK) generator.manual_seed(6148914691236517205 + seed + RANK)
return loader(dataset, return loader(dataset,
batch_size=batch_size, batch_size=batch_size,
shuffle=shuffle and sampler is None, shuffle=shuffle and sampler is None,
utils/segment/dataloaders.py
+3 -2
View File
@@ -37,7 +37,8 @@ def create_dataloader(path,
prefix='', prefix='',
shuffle=False, shuffle=False,
mask_downsample_ratio=1, mask_downsample_ratio=1,
overlap_mask=False): overlap_mask=False,
seed=0):
if rect and shuffle: if rect and shuffle:
LOGGER.warning('WARNING ⚠️ --rect is incompatible with DataLoader shuffle, setting shuffle=False') LOGGER.warning('WARNING ⚠️ --rect is incompatible with DataLoader shuffle, setting shuffle=False')
shuffle = False shuffle = False
@@ -64,7 +65,7 @@ def create_dataloader(path,
sampler = None if rank == -1 else distributed.DistributedSampler(dataset, shuffle=shuffle) sampler = None if rank == -1 else distributed.DistributedSampler(dataset, shuffle=shuffle)
loader = DataLoader if image_weights else InfiniteDataLoader # only DataLoader allows for attribute updates loader = DataLoader if image_weights else InfiniteDataLoader # only DataLoader allows for attribute updates
generator = torch.Generator() generator = torch.Generator()
generator.manual_seed(6148914691236517205 + RANK) generator.manual_seed(6148914691236517205 + seed + RANK)
return loader( return loader(
dataset, dataset,
batch_size=batch_size, batch_size=batch_size,