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
71b59a8d
"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "a0939977a3b3c34c925c565c3fd3dcbe5d09e23c"
Unverified
Commit
71b59a8d
authored
Aug 03, 2020
by
Jintao Lin
Committed by
GitHub
Aug 03, 2020
Browse files
assign `self.warmup_iters` in `before_run` function (#458)
parent
9de04d22
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
3 deletions
+4
-3
mmcv/runner/hooks/lr_updater.py
mmcv/runner/hooks/lr_updater.py
+4
-3
No files found.
mmcv/runner/hooks/lr_updater.py
View file @
71b59a8d
...
@@ -110,13 +110,14 @@ class LrUpdaterHook(Hook):
...
@@ -110,13 +110,14 @@ class LrUpdaterHook(Hook):
group
[
'initial_lr'
]
for
group
in
runner
.
optimizer
.
param_groups
group
[
'initial_lr'
]
for
group
in
runner
.
optimizer
.
param_groups
]
]
def
before_train_epoch
(
self
,
runner
):
if
not
self
.
by_epoch
:
return
if
self
.
warmup_by_epoch
:
if
self
.
warmup_by_epoch
:
epoch_len
=
len
(
runner
.
data_loader
)
epoch_len
=
len
(
runner
.
data_loader
)
self
.
warmup_iters
=
self
.
warmup_epochs
*
epoch_len
self
.
warmup_iters
=
self
.
warmup_epochs
*
epoch_len
def
before_train_epoch
(
self
,
runner
):
if
not
self
.
by_epoch
:
return
self
.
regular_lr
=
self
.
get_regular_lr
(
runner
)
self
.
regular_lr
=
self
.
get_regular_lr
(
runner
)
self
.
_set_lr
(
runner
,
self
.
regular_lr
)
self
.
_set_lr
(
runner
,
self
.
regular_lr
)
...
...
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