"torchvision/transforms/v2/_meta.py" did not exist on "4c049ca3b74c2f93bb2acd952548626aada08fe0"
Commit 242b2d37 authored by Kai Zhang's avatar Kai Zhang Committed by Facebook GitHub Bot
Browse files

Support export_predictors

Summary: As titled.

Reviewed By: newstzpz

Differential Revision: D27074737

fbshipit-source-id: 72f2535fc730a37f5ea8f58aaff88005c28ffc5b
parent 9051f71a
......@@ -228,8 +228,12 @@ class DefaultTask(pl.LightningModule):
return [optim], [{"scheduler": lr_scheduler, "interval": "step"}]
@staticmethod
def build_detection_train_loader(cfg, *args, mapper=None, **kwargs):
return Detectron2GoRunner.build_detection_train_loader(cfg, *args, **kwargs)
def train_dataloader(self):
return Detectron2GoRunner.build_detection_train_loader(self.cfg)
return self.build_detection_train_loader(self.cfg)
def _reset_dataset_evaluators(self):
"""reset validation dataset evaluator to be run in EVAL_PERIOD steps"""
......@@ -273,15 +277,17 @@ class DefaultTask(pl.LightningModule):
dataset_evaluators.append(evaluator)
# TODO: add visualization evaluator
@staticmethod
def build_detection_test_loader(cfg, dataset_name, mapper=None):
return Detectron2GoRunner.build_detection_test_loader(cfg, dataset_name, mapper)
def _evaluation_dataloader(self):
# TODO: Support subsample n images
assert len(self.cfg.DATASETS.TEST)
dataloaders = []
for dataset_name in self.cfg.DATASETS.TEST:
dataloaders.append(
Detectron2GoRunner.build_detection_test_loader(self.cfg, dataset_name)
)
dataloaders.append(self.build_detection_test_loader(self.cfg, dataset_name))
self._reset_dataset_evaluators()
return dataloaders
......
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