Skip to content

Commit 14b5d8e

Browse files
authored
Change all fp16 operation to using MMCV (open-mmlab#3766)
* Change all fp16 operation to using MMCV * fix isort * yapf and docformatter * Revert "yapf and docformatter" This reverts commit dede22f. * yapf
1 parent 7b75383 commit 14b5d8e

34 files changed

+56
-405
lines changed

docs/api.rst

-5
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,6 @@ post_processing
3434
.. automodule:: mmdet.core.post_processing
3535
:members:
3636

37-
fp16
38-
^^^^^^^^^^
39-
.. automodule:: mmdet.core.fp16
40-
:members:
41-
4237
optimizer
4338
^^^^^^^^^^
4439
.. automodule:: mmdet.core.optimizer

mmdet/apis/train.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@
44
import torch
55
from mmcv.parallel import MMDataParallel, MMDistributedDataParallel
66
from mmcv.runner import (HOOKS, DistSamplerSeedHook, EpochBasedRunner,
7-
OptimizerHook, build_optimizer)
7+
Fp16OptimizerHook, OptimizerHook, build_optimizer)
88
from mmcv.utils import build_from_cfg
99

10-
from mmdet.core import DistEvalHook, EvalHook, Fp16OptimizerHook
10+
from mmdet.core import DistEvalHook, EvalHook
1111
from mmdet.datasets import build_dataloader, build_dataset
1212
from mmdet.utils import get_root_logger
1313

mmdet/core/__init__.py

-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from .anchor import * # noqa: F401, F403
22
from .bbox import * # noqa: F401, F403
33
from .evaluation import * # noqa: F401, F403
4-
from .fp16 import * # noqa: F401, F403
54
from .mask import * # noqa: F401, F403
65
from .post_processing import * # noqa: F401, F403
76
from .utils import * # noqa: F401, F403

mmdet/core/fp16/__init__.py

-4
This file was deleted.

mmdet/core/fp16/decorators.py

-164
This file was deleted.

mmdet/core/fp16/hooks.py

-158
This file was deleted.

mmdet/core/fp16/utils.py

-33
This file was deleted.

0 commit comments

Comments
 (0)