extend sft loop with one more step

This commit is contained in:
svlandeg 2025-12-30 09:33:33 +01:00
parent 8f979a8bda
commit de50af6283

View File

@ -167,8 +167,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 % eval_every == 0: