fix: improve handling of cosine hyp

2025-05-23 16:26:31 +07:00
parent af22816363
commit a65348dce4
2 changed files with 5 additions and 4 deletions
--- a/tests/core/test_trainer_builder.py
+++ b/tests/core/test_trainer_builder.py
@@ -216,6 +216,7 @@ class TestHFRLTrainerBuilder:
        assert training_arguments.lr_scheduler_type == "cosine"
        assert training_arguments.warmup_steps == 10
        assert training_arguments.cosine_min_lr_ratio == 0.1
+        assert training_arguments.cosine_constant_lr_ratio == 0.2

        # Other settings
        assert training_arguments.dataloader_num_workers == 1