Commit 66df06ef authored by Peizhao Zhang's avatar Peizhao Zhang Committed by Facebook GitHub Bot
Browse files

Format changes.

Summary:
Format changes.
* [Option] + [Shift] + [F]

Reviewed By: mattcyu1, zhanghang1989, wat3rBro

Differential Revision: D27316555

fbshipit-source-id: 0fc3396eb34d964478cb3551dc73b47412089ccb
parent 4bae056b
...@@ -14,7 +14,12 @@ import d2go.utils.abnormal_checker as abnormal_checker ...@@ -14,7 +14,12 @@ import d2go.utils.abnormal_checker as abnormal_checker
import detectron2.utils.comm as comm import detectron2.utils.comm as comm
import mock import mock
import torch import torch
from d2go.config import CfgNode as CN, CONFIG_SCALING_METHOD_REGISTRY, temp_defrost, get_cfg_diff_table from d2go.config import (
CfgNode as CN,
CONFIG_SCALING_METHOD_REGISTRY,
temp_defrost,
get_cfg_diff_table,
)
from d2go.data.build import build_d2go_train_loader from d2go.data.build import build_d2go_train_loader
from d2go.data.dataset_mappers import build_dataset_mapper from d2go.data.dataset_mappers import build_dataset_mapper
from d2go.data.datasets import inject_coco_datasets, register_dynamic_datasets from d2go.data.datasets import inject_coco_datasets, register_dynamic_datasets
...@@ -34,11 +39,12 @@ from d2go.modeling.quantization import ( ...@@ -34,11 +39,12 @@ from d2go.modeling.quantization import (
setup_qat_model, setup_qat_model,
silicon_qat_build_model_context, silicon_qat_build_model_context,
) )
from d2go.optimizer import build_optimizer_mapper
from d2go.utils.flop_calculator import add_print_flops_callback from d2go.utils.flop_calculator import add_print_flops_callback
from d2go.utils.get_default_cfg import get_default_cfg
from d2go.utils.helper import TensorboardXWriter, D2Trainer
from d2go.utils.misc import get_tensorboard_log_dir from d2go.utils.misc import get_tensorboard_log_dir
from d2go.utils.visualization import DataLoaderVisWrapper, VisualizationEvaluator from d2go.utils.visualization import DataLoaderVisWrapper, VisualizationEvaluator
from d2go.utils.get_default_cfg import get_default_cfg
from d2go.optimizer import build_optimizer_mapper
from detectron2.checkpoint import DetectionCheckpointer, PeriodicCheckpointer from detectron2.checkpoint import DetectionCheckpointer, PeriodicCheckpointer
from detectron2.data import ( from detectron2.data import (
build_detection_test_loader as d2_build_detection_test_loader, build_detection_test_loader as d2_build_detection_test_loader,
...@@ -56,7 +62,6 @@ from detectron2.evaluation import ( ...@@ -56,7 +62,6 @@ from detectron2.evaluation import (
) )
from detectron2.export.caffe2_inference import ProtobufDetectionModel from detectron2.export.caffe2_inference import ProtobufDetectionModel
from detectron2.export.caffe2_modeling import META_ARCH_CAFFE2_EXPORT_TYPE_MAP from detectron2.export.caffe2_modeling import META_ARCH_CAFFE2_EXPORT_TYPE_MAP
from d2go.utils.helper import TensorboardXWriter, D2Trainer
from detectron2.modeling import GeneralizedRCNNWithTTA, build_model from detectron2.modeling import GeneralizedRCNNWithTTA, build_model
from detectron2.solver import ( from detectron2.solver import (
build_lr_scheduler as d2_build_lr_scheduler, build_lr_scheduler as d2_build_lr_scheduler,
...@@ -453,9 +458,11 @@ class Detectron2GoRunner(BaseRunner): ...@@ -453,9 +458,11 @@ class Detectron2GoRunner(BaseRunner):
trainer.register_hooks(trainer_hooks) trainer.register_hooks(trainer_hooks)
trainer.train(start_iter, max_iter) trainer.train(start_iter, max_iter)
if hasattr(self, 'original_cfg'): if hasattr(self, "original_cfg"):
table = get_cfg_diff_table(cfg, self.original_cfg) table = get_cfg_diff_table(cfg, self.original_cfg)
logger.info("GeneralizeRCNN Runner ignoring training config change: \n" + table) logger.info(
"GeneralizeRCNN Runner ignoring training config change: \n" + table
)
trained_cfg = self.original_cfg.clone() trained_cfg = self.original_cfg.clone()
else: else:
trained_cfg = cfg.clone() trained_cfg = cfg.clone()
......
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