From 20f8138146dadb40c6c678ca9b64692a70bfe1d2 Mon Sep 17 00:00:00 2001
From: Ethanfel <ethan.fel@ts-pc.fr>
Date: Sun, 5 Apr 2026 23:45:43 +0200
Subject: [PATCH] chore: show batch_size in training step log

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 nodes/selva_lora_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes/selva_lora_trainer.py b/nodes/selva_lora_trainer.py
index ece535f..2f875f3 100644
--- a/nodes/selva_lora_trainer.py
+++ b/nodes/selva_lora_trainer.py
@@ -504,7 +504,7 @@ class SelvaLoraTrainer:
                 loss_history.append(avg)
                 lr_now = scheduler.get_last_lr()[0]
                 print(f"[LoRA Trainer] step {step:5d}/{steps}  "
-                      f"loss={avg:.4f}  lr={lr_now:.2e}", flush=True)
+                      f"loss={avg:.4f}  lr={lr_now:.2e}  bs={batch_size}", flush=True)
                 running_loss = 0.0
 
                 # Live preview: send updated loss curve to ComfyUI frontend