This commit is contained in:
Sofie Van Landeghem 2026-01-29 11:44:26 +10:00 committed by GitHub
commit ff3363fe5c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -171,8 +171,8 @@ def get_lr_multiplier(it):
# Go!
step = 0
for step in range(num_iterations):
last_step = step == num_iterations - 1
for step in range(num_iterations + 1):
last_step = step == num_iterations
# evaluate the validation loss
if last_step or step % args.eval_every == 0: