Unverified Commit de90c7a2 authored by kira's avatar kira Committed by GitHub
Browse files

[Enhancement] Update the missing warning message for LrUpdaterHook (#2021)

parent e5d9ef52
......@@ -35,7 +35,7 @@ class LrUpdaterHook(Hook):
if warmup not in ['constant', 'linear', 'exp']:
raise ValueError(
f'"{warmup}" is not a supported type for warming up, valid'
' types are "constant" and "linear"')
' types are "constant", "linear" and "exp"')
if warmup is not None:
assert warmup_iters > 0, \
'"warmup_iters" must be a positive integer'
......
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