From 6d80312c8faa616289cf9944c757ef4e5583a2fb Mon Sep 17 00:00:00 2001 From: "limengzhang.vendor" Date: Fri, 10 Jun 2022 13:13:16 +0000 Subject: [PATCH] [Fix] Fix DefaultSampler to InfiniteSampler --- configs/_base_/datasets/ade20k.py | 2 +- configs/_base_/datasets/ade20k_640x640.py | 2 +- configs/_base_/datasets/chase_db1.py | 2 +- configs/_base_/datasets/cityscapes.py | 2 +- configs/_base_/datasets/coco-stuff10k.py | 2 +- configs/_base_/datasets/coco-stuff164k.py | 2 +- configs/_base_/datasets/drive.py | 2 +- configs/_base_/datasets/hrf.py | 2 +- configs/_base_/datasets/isaid.py | 2 +- configs/_base_/datasets/loveda.py | 2 +- configs/_base_/datasets/pascal_context.py | 2 +- configs/_base_/datasets/pascal_context_59.py | 2 +- configs/_base_/datasets/pascal_voc12.py | 2 +- configs/_base_/datasets/potsdam.py | 2 +- configs/_base_/datasets/stare.py | 2 +- configs/_base_/datasets/vaihingen.py | 2 +- 16 files changed, 16 insertions(+), 16 deletions(-) diff --git a/configs/_base_/datasets/ade20k.py b/configs/_base_/datasets/ade20k.py index e6bae5344..f863404ea 100644 --- a/configs/_base_/datasets/ade20k.py +++ b/configs/_base_/datasets/ade20k.py @@ -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, diff --git a/configs/_base_/datasets/ade20k_640x640.py b/configs/_base_/datasets/ade20k_640x640.py index dbcbddf01..ec7c5de89 100644 --- a/configs/_base_/datasets/ade20k_640x640.py +++ b/configs/_base_/datasets/ade20k_640x640.py @@ -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, diff --git a/configs/_base_/datasets/chase_db1.py b/configs/_base_/datasets/chase_db1.py index 5c1be3dd4..8f917e918 100644 --- a/configs/_base_/datasets/chase_db1.py +++ b/configs/_base_/datasets/chase_db1.py @@ -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( diff --git a/configs/_base_/datasets/cityscapes.py b/configs/_base_/datasets/cityscapes.py index 51936fed7..32bcd450f 100644 --- a/configs/_base_/datasets/cityscapes.py +++ b/configs/_base_/datasets/cityscapes.py @@ -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, diff --git a/configs/_base_/datasets/coco-stuff10k.py b/configs/_base_/datasets/coco-stuff10k.py index d36ed4752..b660e6867 100644 --- a/configs/_base_/datasets/coco-stuff10k.py +++ b/configs/_base_/datasets/coco-stuff10k.py @@ -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, diff --git a/configs/_base_/datasets/coco-stuff164k.py b/configs/_base_/datasets/coco-stuff164k.py index eb012eece..9ea2e20a5 100644 --- a/configs/_base_/datasets/coco-stuff164k.py +++ b/configs/_base_/datasets/coco-stuff164k.py @@ -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, diff --git a/configs/_base_/datasets/drive.py b/configs/_base_/datasets/drive.py index f90c90582..bac4a67eb 100644 --- a/configs/_base_/datasets/drive.py +++ b/configs/_base_/datasets/drive.py @@ -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( diff --git a/configs/_base_/datasets/hrf.py b/configs/_base_/datasets/hrf.py index 43fc7a441..b9e6b894a 100644 --- a/configs/_base_/datasets/hrf.py +++ b/configs/_base_/datasets/hrf.py @@ -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( diff --git a/configs/_base_/datasets/isaid.py b/configs/_base_/datasets/isaid.py index cdf426761..2f15fac10 100644 --- a/configs/_base_/datasets/isaid.py +++ b/configs/_base_/datasets/isaid.py @@ -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, diff --git a/configs/_base_/datasets/loveda.py b/configs/_base_/datasets/loveda.py index 349f008df..1df7a4d65 100644 --- a/configs/_base_/datasets/loveda.py +++ b/configs/_base_/datasets/loveda.py @@ -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, diff --git a/configs/_base_/datasets/pascal_context.py b/configs/_base_/datasets/pascal_context.py index d19d2f346..e05c59fef 100644 --- a/configs/_base_/datasets/pascal_context.py +++ b/configs/_base_/datasets/pascal_context.py @@ -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, diff --git a/configs/_base_/datasets/pascal_context_59.py b/configs/_base_/datasets/pascal_context_59.py index 42802e2ff..58cde6cc7 100644 --- a/configs/_base_/datasets/pascal_context_59.py +++ b/configs/_base_/datasets/pascal_context_59.py @@ -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, diff --git a/configs/_base_/datasets/pascal_voc12.py b/configs/_base_/datasets/pascal_voc12.py index 63cf266de..a8611b50a 100644 --- a/configs/_base_/datasets/pascal_voc12.py +++ b/configs/_base_/datasets/pascal_voc12.py @@ -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, diff --git a/configs/_base_/datasets/potsdam.py b/configs/_base_/datasets/potsdam.py index 70da17110..ef7d9ef03 100644 --- a/configs/_base_/datasets/potsdam.py +++ b/configs/_base_/datasets/potsdam.py @@ -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, diff --git a/configs/_base_/datasets/stare.py b/configs/_base_/datasets/stare.py index f18db4889..4c91c8c54 100644 --- a/configs/_base_/datasets/stare.py +++ b/configs/_base_/datasets/stare.py @@ -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( diff --git a/configs/_base_/datasets/vaihingen.py b/configs/_base_/datasets/vaihingen.py index 0a3f7e4ec..f564f5667 100644 --- a/configs/_base_/datasets/vaihingen.py +++ b/configs/_base_/datasets/vaihingen.py @@ -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,