Add invalid_vrs_ids filtering

Summary: As title, with this diff, we verified the dataloading can be reproduced with `num_worker=0` and no data augmentations Differential Revision: D49954475
facebookresearch · Oct 9, 2023 · a6499de · a6499de
1 parent b375c29
commit a6499de
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/d2go/data/build.py b/d2go/data/build.py
@@ -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,
     )