From f651ca030c9546ba99a0909f14690383b178e384 Mon Sep 17 00:00:00 2001 From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com> Date: Wed, 1 Jul 2026 07:37:47 -0400 Subject: [PATCH] fix: convert warmup_ratio to warmup_steps (fix deprecation warning) --- training/scripts/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/scripts/train.py b/training/scripts/train.py index fed7e20..627d90c 100755 --- a/training/scripts/train.py +++ b/training/scripts/train.py @@ -86,7 +86,7 @@ def train(config_path): learning_rate=config["train_params"]["learning_rate"], lr_scheduler_type=config["train_params"]["lr_scheduler_type"], weight_decay=config["train_params"]["weight_decay"], - warmup_ratio=config["train_params"]["warmup_ratio"], + warmup_steps=int(config["train_params"]["warmup_ratio"] * config["train_params"]["num_train_epochs"] * len(dataset["train"]) // config["train_params"]["per_device_train_batch_size"]), logging_steps=config["train_params"]["logging_steps"], save_steps=config["train_params"]["save_steps"], save_total_limit=config["train_params"]["save_total_limit"],