diff options
Diffstat (limited to 'training/optimization.py')
-rw-r--r-- | training/optimization.py | 2 |
1 files changed, 1 insertions, 1 deletions
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( | |||
15 | warmup: Literal["cos", "linear"] = "cos", | 15 | warmup: Literal["cos", "linear"] = "cos", |
16 | annealing: Literal["cos", "half_cos", "linear"] = "cos", | 16 | annealing: Literal["cos", "half_cos", "linear"] = "cos", |
17 | warmup_exp: int = 1, | 17 | warmup_exp: int = 1, |
18 | annealing_exp: int = 2, | 18 | annealing_exp: int = 1, |
19 | min_lr: int = 0.04, | 19 | min_lr: int = 0.04, |
20 | mid_point: int = 0.3, | 20 | mid_point: int = 0.3, |
21 | last_epoch: int = -1 | 21 | last_epoch: int = -1 |