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

4d928927 · Prabhat Roy · GitHub · c1f1e226 · 4d928927
Unverified Commit 4d928927 authored Sep 14, 2021 by Prabhat Roy Committed by GitHub Sep 14, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

references/classification/train.py references/classification/train.py +2 -2

No files found.
--- a/references/classification/train.py
+++ b/references/classification/train.py
@@ -324,8 +324,8 @@ def get_args_parser(add_help=True):
        '--model-ema', action='store_true',
        help='enable tracking Exponential Moving Average of model parameters')
    parser.add_argument(
-        '--model-ema-decay', type=float, default=0.99,
-        help='decay factor for Exponential Moving Average of model parameters(default: 0.99)')
+        '--model-ema-decay', type=float, default=0.9,
+        help='decay factor for Exponential Moving Average of model parameters(default: 0.9)')

    return parser