From 5e5ad1bc20edb2942437ee6f54072b10bbd9d972 Mon Sep 17 00:00:00 2001
From: Elaina Claus <elaina.claus@chtm.me>
Date: Sat, 20 Dec 2025 01:54:42 +0000
Subject: [PATCH] fix log entry for model state saving adjust default batch
 size and passes to be a bit more CPU pytorch friendly

---
 pairwise_compare.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pairwise_compare.py b/pairwise_compare.py
index f9517c0..83f30d8 100755
--- a/pairwise_compare.py
+++ b/pairwise_compare.py
@@ -13,8 +13,8 @@ DEVICE = torch.accelerator.current_accelerator() if torch.accelerator.is_availab
 
 # Valves
 DIMENSIONS = 1
-TRAIN_STEPS = 25000
-TRAIN_BATCHSZ = 16384
+TRAIN_STEPS = 10000
+TRAIN_BATCHSZ = 8192
 TRAIN_PROGRESS = 100
 BATCH_LOWER = -512.0
 BATCH_UPPER = 512.0
@@ -95,7 +95,7 @@ def training_entry():
 
     # embed model dimensions into the model serialization
     torch.save({"state_dict": model.state_dict(), "d": DIMENSIONS}, MODEL_PATH)
-    log.info("Saved PyTorch Model State to model.pth")
+    log.info(f"Saved PyTorch Model State to {MODEL_PATH}")
 
 def infer_entry():
     model_ckpt = torch.load(MODEL_PATH, map_location=DEVICE)