We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7ecc6d8 commit c95812cCopy full SHA for c95812c
references/classification/train.py
@@ -397,8 +397,8 @@ def get_args_parser(add_help=True):
397
'--model-ema-steps', type=int, default=32,
398
help='the number of iterations that controls how often to update the EMA model (default: 32)')
399
parser.add_argument(
400
- '--model-ema-decay', type=float, default=0.99999,
401
- help='decay factor for Exponential Moving Average of model parameters (default: 0.99999)')
+ '--model-ema-decay', type=float, default=0.9999,
+ help='decay factor for Exponential Moving Average of model parameters (default: 0.9999)')
402
403
return parser
404
0 commit comments