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

merge repeat_aug (bug in multi-node?)

pull/1696/head
gaotongxiao 2023-01-23 09:53:21 +08:00
parent e2fd325a95
commit 393506a97c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmocr/datasets/samplers/repeat_aug.py
View File

@ -2,10 +2,11 @@ import math
from typing import Iterator, Optional, Sized from typing import Iterator, Optional, Sized
import torch import torch
from mmcls.registry import DATA_SAMPLERS
from mmengine.dist import get_dist_info, is_main_process, sync_random_seed from mmengine.dist import get_dist_info, is_main_process, sync_random_seed
from torch.utils.data import Sampler from torch.utils.data import Sampler
from mmocr.registry import DATA_SAMPLERS
@DATA_SAMPLERS.register_module() @DATA_SAMPLERS.register_module()
class RepeatAugSampler(Sampler): class RepeatAugSampler(Sampler):