mirror of
https://github.com/karpathy/nanochat.git
synced 2026-07-04 11:59:13 +00:00
extend sft loop with one more step
This commit is contained in:
parent
8f979a8bda
commit
de50af6283
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user