tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix losses

Browse Source
This commit is contained in:
Wing Lian
2026-04-25 08:44:37 +00:00
parent 1b9520cc8b
commit 993db05b3a
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/e2e/test_optimizers.py
View File

@@ -251,7 +251,7 @@ class TestCustomOptimizers(unittest.TestCase):
"adapter": "lora", "adapter": "lora",
"lora_r": 8, "lora_r": 8,
"lora_alpha": 16, "lora_alpha": 16,
"lora_dropout": 0.05, "lora_dropout": 0.0,
"lora_target_linear": True, "lora_target_linear": True,
"val_set_size": 0.1, "val_set_size": 0.1,
"special_tokens": { "special_tokens": {
@@ -264,18 +264,22 @@ class TestCustomOptimizers(unittest.TestCase):
}, },
], ],
"num_epochs": 1, "num_epochs": 1,
"sample_packing": True,
"pad_to_sequence_len": True,
"micro_batch_size": 8, "micro_batch_size": 8,
"gradient_accumulation_steps": 1, "gradient_accumulation_steps": 1,
"output_dir": temp_dir, "output_dir": temp_dir,
"learning_rate": 2e-4, "learning_rate": 1e-4,
"optimizer": "came_pytorch", "optimizer": "came_pytorch",
"adam_beta3": 0.9999, "adam_beta3": 0.9999,
"adam_epsilon2": 1e-16, "adam_epsilon2": 1e-16,
"max_steps": 50, "max_steps": 80,
"warmup_steps": 5,
"logging_steps": 1, "logging_steps": 1,
"lr_scheduler": "cosine", "lr_scheduler": "cosine",
"save_first_step": False, "save_first_step": False,
"use_tensorboard": True, "use_tensorboard": True,
"seed": 42,
} }
) )
@@ -287,10 +291,10 @@ class TestCustomOptimizers(unittest.TestCase):
check_model_output_exists(temp_dir, cfg) check_model_output_exists(temp_dir, cfg)
check_tensorboard_loss_decreased( check_tensorboard_loss_decreased(
temp_dir + "/runs", temp_dir + "/runs",
initial_window=5, initial_window=10,
final_window=5, final_window=10,
max_initial=4.5, max_initial=4.0,
max_final=4.3, max_final=3.0,
) )