"git@developer.sourcefind.cn:OpenDAS/fairscale.git" did not exist on "52d066a2480807d0aba4b8025c0b17aa21dedfff"
Unverified Commit f169fb52 authored by Rui Xu's avatar Rui Xu Committed by GitHub
Browse files

[Feature] Support not to register lr scheduler hook (#774)

* support no lr scheduler hook

* fix lint
parent 3d6032c6
...@@ -344,7 +344,9 @@ class BaseRunner(metaclass=ABCMeta): ...@@ -344,7 +344,9 @@ class BaseRunner(metaclass=ABCMeta):
self.logger.info('resumed epoch %d, iter %d', self.epoch, self.iter) self.logger.info('resumed epoch %d, iter %d', self.epoch, self.iter)
def register_lr_hook(self, lr_config): 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 assert 'policy' in lr_config
policy_type = lr_config.pop('policy') policy_type = lr_config.pop('policy')
# If the type of policy is all in lower case, e.g., 'cyclic', # If the type of policy is all in lower case, e.g., 'cyclic',
......
...@@ -12,7 +12,7 @@ split_before_expression_after_opening_paren = true ...@@ -12,7 +12,7 @@ split_before_expression_after_opening_paren = true
[isort] [isort]
line_length = 79 line_length = 79
multi_line_output = 0 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_first_party = mmcv
known_third_party = addict,cv2,m2r,numpy,onnx,onnxoptimizer,onnxruntime,packaging,pytest,recommonmark,resnet_cifar,tensorrt,torch,torchvision,yaml,yapf 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 no_lines_before = STDLIB,LOCALFOLDER
......
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