mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch 'limengzhang/fix_InfiniteSampler' into 'refactor_dev' 

[Fix] Fix DefaultSampler to InfiniteSampler

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!39
pull/1801/head
zhengmiao 2022-06-10 13:13:17 +00:00
parent 591e56a94f 6d80312c8f
commit 19f39539de
16 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/_base_/datasets/ade20k.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/ade20k_640x640.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/chase_db1.py
View File

@ -25,7 +25,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
type='RepeatDataset',
times=40000,
dataset=dict(

2
configs/_base_/datasets/cityscapes.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=2,
num_workers=2,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/coco-stuff10k.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/coco-stuff164k.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/drive.py
View File

@ -24,7 +24,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
type='RepeatDataset',
times=40000,
dataset=dict(

2
configs/_base_/datasets/hrf.py
View File

@ -24,7 +24,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
type='RepeatDataset',
times=40000,
dataset=dict(

2
configs/_base_/datasets/isaid.py
View File

@ -31,7 +31,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/loveda.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/pascal_context.py
View File

@ -26,7 +26,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/pascal_context_59.py
View File

@ -26,7 +26,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/pascal_voc12.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/potsdam.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,

2
configs/_base_/datasets/stare.py
View File

@ -24,7 +24,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
type='RepeatDataset',
times=40000,
dataset=dict(

2
configs/_base_/datasets/vaihingen.py
View File

@ -23,7 +23,7 @@ train_dataloader = dict(
batch_size=4,
num_workers=4,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
sampler=dict(type='InfiniteSampler', shuffle=True),
dataset=dict(
type=dataset_type,
data_root=data_root,