From adc52fb8821a496bc8d78235bf10466b39df03e0 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 1 Jan 2023 19:19:52 +0100 Subject: Updates --- training/optimization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training/optimization.py') diff --git a/training/optimization.py b/training/optimization.py index 3340544..a79944f 100644 --- a/training/optimization.py +++ b/training/optimization.py @@ -15,7 +15,7 @@ def get_one_cycle_schedule( warmup: Literal["cos", "linear"] = "cos", annealing: Literal["cos", "half_cos", "linear"] = "cos", warmup_exp: int = 1, - annealing_exp: int = 2, + annealing_exp: int = 1, min_lr: int = 0.04, mid_point: int = 0.3, last_epoch: int = -1 -- cgit v1.2.3-54-g00ecf