diff options
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 9aab00c..1b8c597 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -280,7 +280,7 @@ def parse_args(): | |||
| 280 | parser.add_argument( | 280 | parser.add_argument( |
| 281 | "--ema_power", | 281 | "--ema_power", |
| 282 | type=float, | 282 | type=float, |
| 283 | default=7/8 | 283 | default=4/5 |
| 284 | ) | 284 | ) |
| 285 | parser.add_argument( | 285 | parser.add_argument( |
| 286 | "--ema_max_decay", | 286 | "--ema_max_decay", |