Update train.py (#114)

open-mmlab · Mar 1, 2021 · 49f9108 · 49f9108
1 parent bc83cbf
commit 49f9108
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mmtrack/apis/train.py b/mmtrack/apis/train.py
@@ -1,9 +1,8 @@
 import torch
 from mmcv.parallel import MMDataParallel, MMDistributedDataParallel
 from mmcv.runner import (HOOKS, DistSamplerSeedHook, EpochBasedRunner,
-                         OptimizerHook, build_optimizer)
+                         OptimizerHook, build_optimizer, Fp16OptimizerHook)
 from mmcv.utils import build_from_cfg
-from mmdet.core import Fp16OptimizerHook
 from mmdet.datasets import build_dataset
 
 from mmtrack.core import DistEvalHook, EvalHook