diff --git a/mmocr/datasets/samplers/repeat_aug.py b/mmocr/datasets/samplers/repeat_aug.py index 767b58eb..2376fe4a 100644 --- a/mmocr/datasets/samplers/repeat_aug.py +++ b/mmocr/datasets/samplers/repeat_aug.py @@ -2,10 +2,11 @@ import math from typing import Iterator, Optional, Sized import torch -from mmcls.registry import DATA_SAMPLERS from mmengine.dist import get_dist_info, is_main_process, sync_random_seed from torch.utils.data import Sampler +from mmocr.registry import DATA_SAMPLERS + @DATA_SAMPLERS.register_module() class RepeatAugSampler(Sampler):