Skip to content

GitLab

Unverified Commit a47ca7a9 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Fix an import error caused by legacy codes (#186) 

* fix an import error

* fix an import error
parent f4272a88
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 11 deletions
mmcv/runner/__init__.py
View file @ a47ca7a9
...@@ -4,8 +4,7 @@ from .checkpoint import (_load_checkpoint, load_checkpoint, load_state_dict, ...@@ -4,8 +4,7 @@ from .checkpoint import (_load_checkpoint, load_checkpoint, load_state_dict,
from .dist_utils import get_dist_info, init_dist, master_only from .dist_utils import get_dist_info, init_dist, master_only
from .hooks import (CheckpointHook, ClosureHook, DistSamplerSeedHook, Hook, from .hooks import (CheckpointHook, ClosureHook, DistSamplerSeedHook, Hook,
IterTimerHook, LoggerHook, LrUpdaterHook, OptimizerHook, IterTimerHook, LoggerHook, LrUpdaterHook, OptimizerHook,
PaviLoggerHook, TensorboardLoggerHook, TextLoggerHook, TensorboardLoggerHook, TextLoggerHook, WandbLoggerHook)
WandbLoggerHook)
from .log_buffer import LogBuffer from .log_buffer import LogBuffer
from .parallel_test import parallel_test from .parallel_test import parallel_test
from .priority import Priority, get_priority from .priority import Priority, get_priority
...@@ -15,9 +14,9 @@ from .utils import get_host_info, get_time_str, obj_from_dict ...@@ -15,9 +14,9 @@ from .utils import get_host_info, get_time_str, obj_from_dict
__all__ = [ __all__ = [
'Runner', 'LogBuffer', 'Hook', 'CheckpointHook', 'ClosureHook', 'Runner', 'LogBuffer', 'Hook', 'CheckpointHook', 'ClosureHook',
'LrUpdaterHook', 'OptimizerHook', 'IterTimerHook', 'DistSamplerSeedHook', 'LrUpdaterHook', 'OptimizerHook', 'IterTimerHook', 'DistSamplerSeedHook',
'LoggerHook', 'TextLoggerHook', 'PaviLoggerHook', 'TensorboardLoggerHook', 'LoggerHook', 'TextLoggerHook', 'TensorboardLoggerHook', 'WandbLoggerHook',
'WandbLoggerHook', '_load_checkpoint', 'load_state_dict', '_load_checkpoint', 'load_state_dict', 'load_checkpoint', 'weights_to_cpu',
'load_checkpoint', 'weights_to_cpu', 'save_checkpoint', 'parallel_test', 'save_checkpoint', 'parallel_test', 'Priority', 'get_priority',
'Priority', 'get_priority', 'get_host_info', 'get_time_str', 'get_host_info', 'get_time_str', 'obj_from_dict', 'init_dist',
'obj_from_dict', 'init_dist', 'get_dist_info', 'master_only' 'get_dist_info', 'master_only'
] ]
mmcv/runner/hooks/__init__.py
View file @ a47ca7a9
...@@ -3,8 +3,8 @@ from .checkpoint import CheckpointHook ...@@ -3,8 +3,8 @@ from .checkpoint import CheckpointHook
from .closure import ClosureHook from .closure import ClosureHook
from .hook import Hook from .hook import Hook
from .iter_timer import IterTimerHook from .iter_timer import IterTimerHook
from .logger import (LoggerHook, PaviLoggerHook, TensorboardLoggerHook, from .logger import (LoggerHook, TensorboardLoggerHook, TextLoggerHook,
TextLoggerHook, WandbLoggerHook) WandbLoggerHook)
from .lr_updater import LrUpdaterHook from .lr_updater import LrUpdaterHook
from .memory import EmptyCacheHook from .memory import EmptyCacheHook
from .optimizer import OptimizerHook from .optimizer import OptimizerHook
...@@ -13,6 +13,5 @@ from .sampler_seed import DistSamplerSeedHook ...@@ -13,6 +13,5 @@ from .sampler_seed import DistSamplerSeedHook
__all__ = [ __all__ = [
'Hook', 'CheckpointHook', 'ClosureHook', 'LrUpdaterHook', 'OptimizerHook', 'Hook', 'CheckpointHook', 'ClosureHook', 'LrUpdaterHook', 'OptimizerHook',
'IterTimerHook', 'DistSamplerSeedHook', 'EmptyCacheHook', 'LoggerHook', 'IterTimerHook', 'DistSamplerSeedHook', 'EmptyCacheHook', 'LoggerHook',
'TextLoggerHook', 'PaviLoggerHook', 'TensorboardLoggerHook', 'TextLoggerHook', 'TensorboardLoggerHook', 'WandbLoggerHook'
'WandbLoggerHook'
] ]
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