Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
MMCV
Commits
f169fb52
Unverified
Commit
f169fb52
authored
Jan 09, 2021
by
Rui Xu
Committed by
GitHub
Jan 09, 2021
Browse files
[Feature] Support not to register lr scheduler hook (#774)
* support no lr scheduler hook * fix lint
parent
3d6032c6
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
2 deletions
+4
-2
mmcv/runner/base_runner.py
mmcv/runner/base_runner.py
+3
-1
setup.cfg
setup.cfg
+1
-1
No files found.
mmcv/runner/base_runner.py
View file @
f169fb52
...
...
@@ -344,7 +344,9 @@ class BaseRunner(metaclass=ABCMeta):
self
.
logger
.
info
(
'resumed epoch %d, iter %d'
,
self
.
epoch
,
self
.
iter
)
def
register_lr_hook
(
self
,
lr_config
):
if
isinstance
(
lr_config
,
dict
):
if
lr_config
is
None
:
return
elif
isinstance
(
lr_config
,
dict
):
assert
'policy'
in
lr_config
policy_type
=
lr_config
.
pop
(
'policy'
)
# If the type of policy is all in lower case, e.g., 'cyclic',
...
...
setup.cfg
View file @
f169fb52
...
...
@@ -12,7 +12,7 @@ split_before_expression_after_opening_paren = true
[isort]
line_length = 79
multi_line_output = 0
known_standard_library = pkg_resources,setuptools
known_standard_library = pkg_resources,setuptools
,logging,os,warnings,abc
known_first_party = mmcv
known_third_party = addict,cv2,m2r,numpy,onnx,onnxoptimizer,onnxruntime,packaging,pytest,recommonmark,resnet_cifar,tensorrt,torch,torchvision,yaml,yapf
no_lines_before = STDLIB,LOCALFOLDER
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment