From a1b8327085ddeab589be074d7e9df4291aba1210 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Wed, 1 Mar 2023 12:34:42 +0100
Subject: Update

---
 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 6c9a35d..7d8d55a 100644
--- a/training/optimization.py
+++ b/training/optimization.py
@@ -113,7 +113,7 @@ def get_scheduler(
 ):
     num_training_steps_per_epoch = math.ceil(
         num_training_steps_per_epoch / gradient_accumulation_steps
-    ) * gradient_accumulation_steps
+    )  # * gradient_accumulation_steps
     num_training_steps = train_epochs * num_training_steps_per_epoch
     num_warmup_steps = warmup_epochs * num_training_steps_per_epoch
 
-- 
cgit v1.2.3-54-g00ecf