From 014084a65602da747436afe425bb727f61986206 Mon Sep 17 00:00:00 2001 From: gpu-poor Date: Sun, 1 Mar 2026 13:16:18 +0530 Subject: [PATCH] undoing some changes --- nanochat/loss_eval.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/nanochat/loss_eval.py b/nanochat/loss_eval.py index c983601..5a556e6 100644 --- a/nanochat/loss_eval.py +++ b/nanochat/loss_eval.py @@ -28,8 +28,7 @@ def evaluate_bpb(model, batches, steps, token_bytes): total_nats = torch.tensor(0.0, dtype=torch.float32, device=model.get_device()) total_bytes = torch.tensor(0, dtype=torch.int64, device=model.get_device()) batch_iter = iter(batches) - for step in range(steps): - print(f"\reval {step+1}/{steps}", end="", flush=True) + for _ in range(steps): x, y = next(batch_iter) loss2d = model(x, y, loss_reduction='none') # (B, T) loss2d = loss2d.view(-1) # flatten @@ -52,7 +51,6 @@ def evaluate_bpb(model, batches, steps, token_bytes): num_bytes2d = token_bytes[y] total_nats += (loss2d * (num_bytes2d > 0)).sum() total_bytes += num_bytes2d.sum() - print() # newline after progress # sum reduce across all ranks world_size = dist.get_world_size() if dist.is_initialized() else 1 if world_size > 1: