Skip to content

GitLab

Unverified Commit 4d928927 authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Updated model-ema-decay default to 0.9 (#4407)

parent c1f1e226
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
references/classification/train.py
View file @ 4d928927
...@@ -324,8 +324,8 @@ def get_args_parser(add_help=True): ...@@ -324,8 +324,8 @@ def get_args_parser(add_help=True):
'--model-ema', action='store_true', '--model-ema', action='store_true',
help='enable tracking Exponential Moving Average of model parameters') help='enable tracking Exponential Moving Average of model parameters')
parser.add_argument( parser.add_argument(
'--model-ema-decay', type=float, default=0.99, '--model-ema-decay', type=float, default=0.9,
help='decay factor for Exponential Moving Average of model parameters(default: 0.99)') help='decay factor for Exponential Moving Average of model parameters(default: 0.9)')
return parser return parser
......
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