From 393506a97cbe6d75ad1f28611ea10eba6b8fa4b3 Mon Sep 17 00:00:00 2001 From: gaotongxiao Date: Mon, 23 Jan 2023 09:53:21 +0800 Subject: [PATCH] merge repeat_aug (bug in multi-node?) --- mmocr/datasets/samplers/repeat_aug.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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):