From d0b81eda0eb0392442c6700fd11a449d9f95d5c0 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 2 Jan 2023 11:28:42 +0100 Subject: Fix --- training/optimization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/optimization.py b/training/optimization.py index cb7f088..14c2bd5 100644 --- a/training/optimization.py +++ b/training/optimization.py @@ -78,7 +78,7 @@ def get_one_cycle_schedule( ] else: phases = [ - OneCyclePhase(0, thresh_up, 0, 1, warmup_func), + OneCyclePhase(0, thresh_up, min_lr, 1, warmup_func), OneCyclePhase(thresh_up, num_training_steps, 0, 1, anneal_func), ] -- cgit v1.2.3-54-g00ecf