Update mmdet/apis/train.py

mmdet/apis/train.py

@@ -76,7 +76,12 @@ def train_detector(model,
     ]
 
     # build optimizer
-    optimizer = build_optimizer(model, cfg.optimizer)
+    distiller_cfg = cfg.get('distiller',None)
+    if distiller_cfg is None:
+        optimizer = build_optimizer(model, cfg.optimizer)
+    else:
+        #optimizer = build_optimizer(model.module.base_parameters(), cfg.optimizer)
+        optimizer = build_optimizer(model.base_parameters(), cfg.optimizer)
 
     # use apex fp16 optimizer
     if cfg.optimizer_config.get("type", None) and cfg.optimizer_config["type"] == "DistOptimizerHook":