From 4bcc3bb698b802766852c3ef1003f91d589f7b66 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Fri, 21 Nov 2025 13:19:45 +0100 Subject: [PATCH] clarify comment --- scripts/mid_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/mid_train.py b/scripts/mid_train.py index ebb6c42..60c7bbc 100644 --- a/scripts/mid_train.py +++ b/scripts/mid_train.py @@ -112,7 +112,7 @@ val_dataset = TaskMixture([ # DataLoader is defined here, it emits inputs, targets : 2D tensors of shape (device_batch_size, max_seq_len) # A big problem is that we don't know the final num_iterations in advance. So we create # these two global variables and update them from within the data generator. -last_step = False # we will toggle this to True when we reach the end of the dataset +last_step = False # we will toggle this to True when we reach the end of the training dataset approx_progress = 0.0 # will go from 0 to 1 over the course of the epoch def mid_data_generator(split): global last_step, approx_progress