From 000ca89301ba8fb9a58ab718b885d567cfbb4ca1 Mon Sep 17 00:00:00 2001 From: "U. I. I. Derbashi" <32719218+uderbashi@users.noreply.github.com> Date: Thu, 14 May 2026 15:48:28 +0200 Subject: [PATCH] Studio: Passing batch size for eval (#5168) * add eval batch size * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Roland Tannous <115670425+rolandtannous@users.noreply.github.com> Co-authored-by: Lee Jackson <130007945+Imagineer99@users.noreply.github.com> --- studio/backend/core/training/trainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/studio/backend/core/training/trainer.py b/studio/backend/core/training/trainer.py index a3f063694f..62f1e23e60 100644 --- a/studio/backend/core/training/trainer.py +++ b/studio/backend/core/training/trainer.py @@ -3208,6 +3208,9 @@ class UnslothTrainer: if eval_steps_val > 0: config_args["eval_strategy"] = "steps" config_args["eval_steps"] = eval_steps_val + config_args["per_device_eval_batch_size"] = config_args[ + "per_device_train_batch_size" + ] logger.info( f"✅ Evaluation enabled: eval_steps={eval_steps_val} (fraction of total steps)\n" )