chore: show batch_size in training step log

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-05 23:45:43 +02:00
parent 09b3b94ddd
commit 20f8138146
1 changed files with 1 additions and 1 deletions
@@ -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