Fix log off by 1

This commit is contained in:
Muhammad Rizqi Nur
2022-10-28 20:48:08 +07:00
parent 737eb28fac
commit 9ceef81f77
3 changed files with 20 additions and 18 deletions

View File

@@ -52,7 +52,7 @@ class LearnRateScheduler:
self.finished = False
def apply(self, optimizer, step_number):
if step_number <= self.end_step:
if step_number < self.end_step:
return
try: