Skip to content
This repository was archived by the owner on Jan 22, 2025. It is now read-only.

Commit 4b4d52e

Browse files
jiaxuzhu92facebook-github-bot
authored andcommitted
Add invalid_vrs_ids filtering
Summary: X-link: facebookresearch/detectron2#5108 As title, with this diff, we verified the dataloading can be reproduced with `num_worker=2` and with RC6 data augmentations Differential Revision: D49954475
1 parent b375c29 commit 4b4d52e

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

d2go/data/build.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,14 +159,15 @@ def build_weighted_detection_train_loader(
159159
)
160160
)
161161

162-
sampler = RepeatFactorTrainingSampler(repeat_factors)
162+
sampler = RepeatFactorTrainingSampler(repeat_factors, seed=cfg.SEED)
163163

164164
return build_batch_data_loader(
165165
dataset,
166166
sampler,
167167
cfg.SOLVER.IMS_PER_BATCH,
168168
aspect_ratio_grouping=cfg.DATALOADER.ASPECT_RATIO_GROUPING,
169169
num_workers=cfg.DATALOADER.NUM_WORKERS,
170+
seed=cfg.SEED,
170171
)
171172

172173

0 commit comments

Comments
 (0)