From f98f5a2611338c93bc6a63b5d608eafb44250bf9 Mon Sep 17 00:00:00 2001
From: Christian Risi <75698846+CnF-Gris@users.noreply.github.com>
Date: Sun, 12 Oct 2025 01:16:09 +0200
Subject: [PATCH] Fixed misprint in task 3

---
 Playgrounds/nanosocrates-train-experiment-2.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Playgrounds/nanosocrates-train-experiment-2.py b/Playgrounds/nanosocrates-train-experiment-2.py
index 450630f..71844f4 100644
--- a/Playgrounds/nanosocrates-train-experiment-2.py
+++ b/Playgrounds/nanosocrates-train-experiment-2.py
@@ -46,7 +46,7 @@ NUMBER_OF_BLOCKS = 4
 MAX_EPOCHS = int(3e3)
 PRETRAIN_EPOCHS = int(300)
 WARMUP_EPOCHS = int(1e3)
-MINI_BATCH_SIZE = 300
+MINI_BATCH_SIZE = 80
 VALIDATION_STEPS = 5
 CHECKPOINT_STEPS = VALIDATION_STEPS * 4
 PATIENCE = 4
@@ -185,7 +185,7 @@ while current_epoch < MAX_EPOCHS:
 
             pred_logits = ENCODER_ONLY((enc_x, enc_x_pad))
             pred_logits = pred_logits.permute(0, 2, 1)
-            print(torch.max(tgt))
+            # print(torch.max(tgt))
             loss: torch.Tensor = encoder_ce(pred_logits, tgt)
 
             loss.backward()