feat:lvis minival

LutingWang · Oct 16, 2024 · 65685d6 · 65685d6
1 parent 4d47a7c
commit 65685d6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/configs/dp/datasets/lvis.py b/configs/dp/datasets/lvis.py
@@ -69,7 +69,7 @@
     dataset=dict(
         type=dataset_type,
         data_root=data_root,
-        ann_file='annotations/lvis_v1_val.1203.json',
+        ann_file='annotations/lvis_v1_minival.1203.json',
         data_prefix=dict(img=''),
         test_mode=True,
         pipeline=test_pipeline,
@@ -80,7 +80,7 @@
 
 val_evaluator = dict(
     type='LVISMetric',
-    ann_file=data_root + 'annotations/lvis_v1_val.1203.json',
+    ann_file=data_root + 'annotations/lvis_v1_minival.1203.json',
     metric=['bbox'],
     backend_args=None,
 )

diff --git a/oadp/dp/runners.py b/oadp/dp/runners.py
@@ -19,6 +19,8 @@ def from_cfg(cls, config: Config, **kwargs) -> Runner:  # noqa: E501 pylint: dis
         if Store.DRY_RUN:
             config.train_dataloader.num_workers = 0
             config.train_dataloader.persistent_workers = False
+            config.val_dataloader.num_workers = 0
+            config.val_dataloader.persistent_workers = False
 
         if config.pop('autocast', False):
             config.optim_wrapper.type = 'AmpOptimWrapper'