Commit d37031b4 authored by Kai Zhang's avatar Kai Zhang Committed by Facebook GitHub Bot
Browse files

allow overriding get_mapper

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/63

As titled.

Reviewed By: zhanghang1989

Differential Revision: D27945627

fbshipit-source-id: 58180ace6ab327a5eb025f3227ae6c22601a3580
parent b87ee53a
...@@ -313,6 +313,10 @@ class DefaultTask(pl.LightningModule): ...@@ -313,6 +313,10 @@ class DefaultTask(pl.LightningModule):
cfg=cfg, dataset_name=dataset_name, output_folder=output_folder cfg=cfg, dataset_name=dataset_name, output_folder=output_folder
) )
@staticmethod
def get_mapper(cfg, is_train):
return Detectron2GoRunner.get_mapper(cfg, is_train)
@staticmethod @staticmethod
def build_detection_train_loader(cfg, *args, mapper=None, **kwargs): def build_detection_train_loader(cfg, *args, mapper=None, **kwargs):
return Detectron2GoRunner.build_detection_train_loader(cfg, *args, **kwargs) return Detectron2GoRunner.build_detection_train_loader(cfg, *args, **kwargs)
......
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