Commit 2d328adb authored by Peizhao Zhang's avatar Peizhao Zhang Committed by Facebook GitHub Bot
Browse files

Use build_model in d2go in default_runner instead of the version in d2.

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

Use build_model in d2go in default_runner instead of the version in d2.

Differential Revision: D35535568

fbshipit-source-id: e5e2de63787d21a3c51ed3d689f9058c5f4518b3
parent dc5ed4f5
...@@ -23,7 +23,7 @@ from d2go.data.utils import ( ...@@ -23,7 +23,7 @@ from d2go.data.utils import (
update_cfg_if_using_adhoc_dataset, update_cfg_if_using_adhoc_dataset,
) )
from d2go.export.d2_meta_arch import patch_d2_meta_arch from d2go.export.d2_meta_arch import patch_d2_meta_arch
from d2go.modeling import kmeans_anchors, model_ema from d2go.modeling import build_model, kmeans_anchors, model_ema
from d2go.modeling.model_freezing_utils import freeze_matched_bn, set_requires_grad from d2go.modeling.model_freezing_utils import freeze_matched_bn, set_requires_grad
from d2go.optimizer import build_optimizer_mapper from d2go.optimizer import build_optimizer_mapper
from d2go.quantization.modeling import QATCheckpointer, QATHook, setup_qat_model from d2go.quantization.modeling import QATCheckpointer, QATHook, setup_qat_model
...@@ -48,7 +48,7 @@ from detectron2.evaluation import ( ...@@ -48,7 +48,7 @@ from detectron2.evaluation import (
RotatedCOCOEvaluator, RotatedCOCOEvaluator,
verify_results, verify_results,
) )
from detectron2.modeling import build_model, GeneralizedRCNNWithTTA from detectron2.modeling import GeneralizedRCNNWithTTA
from detectron2.solver import build_lr_scheduler as d2_build_lr_scheduler from detectron2.solver import build_lr_scheduler as d2_build_lr_scheduler
from detectron2.utils.events import CommonMetricPrinter, JSONWriter from detectron2.utils.events import CommonMetricPrinter, JSONWriter
from detectron2.utils.registry import Registry from detectron2.utils.registry import Registry
......
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