"vscode:/vscode.git/clone" did not exist on "3efb66213bbfffcc4aa2c8241ad0f719c875897f"
Commit bc1939eb authored by Ajinkya Deogade's avatar Ajinkya Deogade Committed by Facebook GitHub Bot
Browse files

Optimizer: create a separate buck target

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

Create buck target for the optimizer.
Other than creating TARGETS, replaced `from d2go.optimizer import build_optimizer_mapper` with `from d2go.optimizer.build import build_optimizer_mapper`

Reviewed By: tglik

Differential Revision: D45912075

fbshipit-source-id: e478783a9ec16d4573d6365e5567e8d2ed72eb06
parent 1950242a
......@@ -28,7 +28,7 @@ from d2go.evaluation.evaluator import inference_on_dataset
from d2go.modeling import ema, kmeans_anchors
from d2go.modeling.api import build_d2go_model
from d2go.modeling.model_freezing_utils import freeze_matched_bn, set_requires_grad
from d2go.optimizer import build_optimizer_mapper
from d2go.optimizer.build import build_optimizer_mapper
from d2go.quantization.modeling import QATHook, setup_qat_model
from d2go.runner.config_defaults import (
get_base_runner_default_cfg,
......
......@@ -14,7 +14,7 @@ from d2go.data.datasets import inject_coco_datasets, register_dynamic_datasets
from d2go.data.utils import update_cfg_if_using_adhoc_dataset
from d2go.modeling.api import build_meta_arch
from d2go.modeling.model_freezing_utils import set_requires_grad
from d2go.optimizer import build_optimizer_mapper
from d2go.optimizer.build import build_optimizer_mapper
from d2go.runner.api import RunnerV2Mixin
from d2go.runner.callbacks.quantization import maybe_prepare_for_quantization, PREPARED
from d2go.runner.default_runner import (
......
......@@ -7,7 +7,7 @@ import unittest
import d2go.runner.default_runner as default_runner
import torch
from d2go.optimizer import build_optimizer_mapper
from d2go.optimizer.build import build_optimizer_mapper
from d2go.utils.testing import helper
......
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