"src/git@developer.sourcefind.cn:OpenDAS/lmdeploy.git" did not exist on "bb6f8060e3301d0e5c47b9f19b1d983d721e198e"
Unverified Commit 5f611bbb authored by lizz's avatar lizz Committed by GitHub
Browse files

Update lr_updater.py (#334)

parent 6988de82
...@@ -25,8 +25,7 @@ class LrUpdaterHook(Hook): ...@@ -25,8 +25,7 @@ class LrUpdaterHook(Hook):
warmup=None, warmup=None,
warmup_iters=0, warmup_iters=0,
warmup_ratio=0.1, warmup_ratio=0.1,
warmup_by_epoch=False, warmup_by_epoch=False):
**kwargs):
# validate the "warmup" argument # validate the "warmup" argument
if warmup is not None: if warmup is not None:
if warmup not in ['constant', 'linear', 'exp']: if warmup not in ['constant', 'linear', 'exp']:
......
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