From 807a56bdfcdc34b8558a432038b393628ca19b4a Mon Sep 17 00:00:00 2001 From: Salman Mohammadi Date: Mon, 3 Nov 2025 12:32:59 +0000 Subject: [PATCH] nit --- scripts/base_train.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/scripts/base_train.py b/scripts/base_train.py index a57cd42..925f7ce 100644 --- a/scripts/base_train.py +++ b/scripts/base_train.py @@ -202,8 +202,8 @@ for step in range(num_iterations + 1): model.train() # once in a while: estimate the CORE metric (all ranks participate) - # use the eval-compiled model which as the training-compiled model - # has been specialized for a fixed input shape + # use the eval-compiled model as the training-compiled model has + # been specialized for a fixed input shape results = {} if core_metric_every > 0 and (last_step or (step > 0 and step % core_metric_every == 0)): model.eval() @@ -220,8 +220,8 @@ for step in range(num_iterations + 1): model.train() # once in a while: sample from the model (only on master process) - # use the eval-compiled model which as the training-compiled model - # has been specialized for a fixed input shape + # use the eval-compiled model as the training-compiled model has + # been specialized for a fixed input shape if master_process and (last_step or (step > 0 and step % sample_every == 0)): model.eval() prompts = [