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

more train steps

Browse Source
This commit is contained in:
Wing Lian
2026-04-25 02:17:48 +00:00
parent f77408a3d0
commit 1b9520cc8b
8 changed files with 105 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/e2e/test_falcon.py
View File

@@ -166,17 +166,19 @@ class TestFalcon(unittest.TestCase):
},
],
"num_epochs": 2,
"sample_packing": True,
"pad_to_sequence_len": True,
"micro_batch_size": 4,
"gradient_accumulation_steps": 1,
"output_dir": temp_dir,
"learning_rate": 2e-4,
"learning_rate": 5e-4,
"optimizer": "adamw_torch_fused",
"lr_scheduler": "cosine",
"max_steps": 50,
"max_steps": 80,
"warmup_steps": 5,
"logging_steps": 1,
"save_steps": 50,
"eval_steps": 50,
"save_steps": 80,
"eval_steps": 80,
"bf16": "auto",
"save_first_step": False,
"use_tensorboard": True,
@@ -192,8 +194,8 @@ class TestFalcon(unittest.TestCase):
check_model_output_exists(temp_dir, cfg)
check_tensorboard_loss_decreased(
temp_dir + "/runs",
initial_window=5,
final_window=5,
initial_window=10,
final_window=10,
max_initial=5.0,
max_final=4.7,
)