This commit is contained in:
Sofie Van Landeghem 2026-01-28 14:21:55 -06:00 committed by GitHub
commit d5c10f1b31
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -172,8 +172,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: