Skip to content

Commit c95812c

Browse files
committed
Adjust ema decay value
1 parent 7ecc6d8 commit c95812c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

references/classification/train.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -397,8 +397,8 @@ def get_args_parser(add_help=True):
397397
'--model-ema-steps', type=int, default=32,
398398
help='the number of iterations that controls how often to update the EMA model (default: 32)')
399399
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)')
400+
'--model-ema-decay', type=float, default=0.9999,
401+
help='decay factor for Exponential Moving Average of model parameters (default: 0.9999)')
402402

403403
return parser
404404

0 commit comments

Comments
 (0)