Set a seed of generator with an option for more randomness when training several models with different seeds (#10486)
* 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:
1 parent
b2a0f1cdc5
commit
10e93d295f
3 files changed
+8
-5
No files matched your search
@@ -115,7 +115,8 @@ def create_dataloader(path,
|
||||
image_weights=False,
|
||||
quad=False,
|
||||
prefix='',
|
||||
shuffle=False):
|
||||
shuffle=False,
|
||||
seed=0):
|
||||
if rect and shuffle:
|
||||
LOGGER.warning('WARNING ⚠️ --rect is incompatible with DataLoader shuffle, setting shuffle=False')
|
||||
shuffle = False
|
||||
@@ -140,7 +141,7 @@ def create_dataloader(path,
|
||||
sampler = None if rank == -1 else distributed.DistributedSampler(dataset, shuffle=shuffle)
|
||||
loader = DataLoader if image_weights else InfiniteDataLoader # only DataLoader allows for attribute updates
|
||||
generator = torch.Generator()
|
||||
generator.manual_seed(6148914691236517205 + RANK)
|
||||
generator.manual_seed(6148914691236517205 + seed + RANK)
|
||||
return loader(dataset,
|
||||
batch_size=batch_size,
|
||||
shuffle=shuffle and sampler is None,
|
||||
|
||||
Reference in new issue
Block a user