From 2c3dcbb1bc3982b507c3b30b5deed5aedd2160b2 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 6 Jan 2023 18:10:19 +0100 Subject: Relaxed ema_power for TI --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(): parser.add_argument( "--ema_power", type=float, - default=7/8 + default=4/5 ) parser.add_argument( "--ema_max_decay", -- cgit v1.2.3-70-g09d2