diff --git a/Playgrounds/nanosocrates-train-experiment-2.py b/Playgrounds/nanosocrates-train-experiment-2.py
index 8eeda63..e72e41c 100644
--- a/Playgrounds/nanosocrates-train-experiment-2.py
+++ b/Playgrounds/nanosocrates-train-experiment-2.py
@@ -130,7 +130,7 @@ decoder_only_scheduler = Transformer.WarmupLR(
     decoder_only_optim, WARMUP_EPOCHS, EMBEDDED_SIZE, last_epoch=CURRENT_EPOCH
 )
 
-current_epoch = CURRENT_EPOCH
+current_epoch = CURRENT_EPOCH + 2
 patience = 0