Skip to content

Commit

Permalink
Add invalid_vrs_ids filtering
Browse files Browse the repository at this point in the history
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
  • Loading branch information
jiaxuzhu92 authored and facebook-github-bot committed Oct 10, 2023
1 parent b375c29 commit 4b4d52e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion d2go/data/build.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,14 +159,15 @@ def build_weighted_detection_train_loader(
)
)

sampler = RepeatFactorTrainingSampler(repeat_factors)
sampler = RepeatFactorTrainingSampler(repeat_factors, seed=cfg.SEED)

return build_batch_data_loader(
dataset,
sampler,
cfg.SOLVER.IMS_PER_BATCH,
aspect_ratio_grouping=cfg.DATALOADER.ASPECT_RATIO_GROUPING,
num_workers=cfg.DATALOADER.NUM_WORKERS,
seed=cfg.SEED,
)


Expand Down

0 comments on commit 4b4d52e

Please sign in to comment.