Commit afbd4558 authored by Yuxin Wu's avatar Yuxin Wu Committed by Facebook GitHub Bot
Browse files

DPE d2go async loader

Reviewed By: newstzpz

Differential Revision: D31209906

fbshipit-source-id: 0be4e3c1db623e3c1fba8ba4259840d34192a77e
parent b6c93a74
...@@ -6,7 +6,7 @@ import logging ...@@ -6,7 +6,7 @@ import logging
import os import os
from collections import OrderedDict from collections import OrderedDict
from functools import lru_cache from functools import lru_cache
from typing import Type, Optional, List from typing import Type, Optional, List, Union
import d2go.utils.abnormal_checker as abnormal_checker import d2go.utils.abnormal_checker as abnormal_checker
import detectron2.utils.comm as comm import detectron2.utils.comm as comm
...@@ -494,7 +494,9 @@ class Detectron2GoRunner(BaseRunner): ...@@ -494,7 +494,9 @@ class Detectron2GoRunner(BaseRunner):
return {"model_final": trained_cfg} return {"model_final": trained_cfg}
@classmethod @classmethod
def build_detection_test_loader(cls, cfg, dataset_name, mapper=None): def build_detection_test_loader(
cls, cfg, dataset_name: Union[str, List[str]], mapper=None
):
logger.info( logger.info(
"Building detection test loader for dataset: {} ...".format(dataset_name) "Building detection test loader for dataset: {} ...".format(dataset_name)
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment