TRAIN_DATASET = "angie-chen55/python-github-code" TRAIN_BASE_MODEL = "meta-llama/Llama-3.2-1B"

TRAIN_MODEL_HUB_NAME = "steve329" TRAIN_TUNED_MODEL_NAME = "test_fine_llama3_qlora_all_6" TRAIN_PEFT_LAYERS = "q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj" TRAIN_LORA_ALPHA = 16 TRAIN_LORA_DROPOUT = 0.05 TRAIN_LORA_R = 8

TRAIN_SAMPLES = 5000 TRAIN_VAL_SAMPLES = 500 TRAIN_SEQ_LEN = 1024 TRAIN_MAX_STEPS = 400 TRAIN_VAL_STEPS = 80 TRAIN_SAVE_STEPS = 1000 TRAIN_LOG_STEPS = 10 TRAIN_TRAIN_BSIZE = 1 TRAIN_VAL_BSIZE = 1 TRAIN_LR = 5e-5 TRAIN_LR_TYPE = "cosine" TRAIN_WARMUP_STEPS = 0 TRAIN_FP16 = True TRAIN_BF16 = False TRAIN_ACCU_STEPS = 1 TRAIN_WEIGHT_DECAY = 0.01

TEST_NUM_SAMPLES_PER_TASK = 15 TEST_NUM_TASK_IDS = 50 TEST_MAX_NEW_TOEKNS = 512

TEST_TEMPERATURE = 0.2 TEST_TOP_K = 50 TEST_TOP_P = 0.95 TEST_DO_SAMPLE = True TEST_REPETITION_PENALTY = 1.0

--- tuning --- trainable params: 5,636,096 || all params: 1,241,450,496 || trainable%: 0.4540 Resolving data files: 100%|███████████████████████████████████████████████████████████████| 108/108 [00:00<00:00, 249.25it/s] max_steps is given, it will override any value given in num_train_epochs 0%| | 0/400 [00:00<?, ?it/s]The input hidden states seems to be silently casted in float32, this might be related to the fact you have upcasted embedding or layer norm layers in float32. We will cast back the input in torch.float16. {'loss': 1.2884, 'grad_norm': 0.7126919627189636, 'learning_rate': 4.99229333433282e-05, 'epoch': 0.03}
{'loss': 1.5233, 'grad_norm': 1.0026756525039673, 'learning_rate': 4.9692208514878444e-05, 'epoch': 0.05}
{'loss': 1.4726, 'grad_norm': 0.930830717086792, 'learning_rate': 4.9309248009941914e-05, 'epoch': 0.07}
{'loss': 1.4719, 'grad_norm': 1.027666687965393, 'learning_rate': 4.877641290737884e-05, 'epoch': 0.1}
{'loss': 1.3395, 'grad_norm': 0.8995978832244873, 'learning_rate': 4.8096988312782174e-05, 'epoch': 0.12}
{'loss': 1.5036, 'grad_norm': 1.188391923904419, 'learning_rate': 4.72751631047092e-05, 'epoch': 0.15}
{'loss': 1.5441, 'grad_norm': 0.7593730092048645, 'learning_rate': 4.6316004108852305e-05, 'epoch': 0.17}
{'loss': 1.2875, 'grad_norm': 1.0513273477554321, 'learning_rate': 4.522542485937369e-05, 'epoch': 0.2}
{'eval_loss': 1.347599744796753, 'eval_runtime': 318.23, 'eval_samples_per_second': 3.466, 'eval_steps_per_second': 3.466, 'epoch': 0.2} {'loss': 1.6887, 'grad_norm': 1.1072134971618652, 'learning_rate': 4.401014914000078e-05, 'epoch': 0.23}
{'loss': 1.5142, 'grad_norm': 1.014134168624878, 'learning_rate': 4.267766952966369e-05, 'epoch': 0.25}
{'loss': 1.4802, 'grad_norm': 1.056219220161438, 'learning_rate': 4.123620120825459e-05, 'epoch': 0.28}
{'loss': 1.0564, 'grad_norm': 0.8529524207115173, 'learning_rate': 3.969463130731183e-05, 'epoch': 0.3}
{'loss': 1.4779, 'grad_norm': 1.8953529596328735, 'learning_rate': 3.8062464117898724e-05, 'epoch': 0.33}
{'loss': 1.223, 'grad_norm': 0.8968033790588379, 'learning_rate': 3.634976249348867e-05, 'epoch': 0.35}
{'loss': 1.6358, 'grad_norm': 1.1496285200119019, 'learning_rate': 3.456708580912725e-05, 'epoch': 0.38}
{'loss': 1.7058, 'grad_norm': 0.9135394096374512, 'learning_rate': 3.272542485937369e-05, 'epoch': 0.4}
{'eval_loss': 1.339319109916687, 'eval_runtime': 317.3955, 'eval_samples_per_second': 3.475, 'eval_steps_per_second': 3.475, 'epoch': 0.4} {'loss': 1.5306, 'grad_norm': 1.1340606212615967, 'learning_rate': 3.083613409639764e-05, 'epoch': 0.42}
{'loss': 1.3567, 'grad_norm': 1.5782934427261353, 'learning_rate': 2.8910861626005776e-05, 'epoch': 0.45}
{'loss': 1.3825, 'grad_norm': 0.9451084136962891, 'learning_rate': 2.6961477393196126e-05, 'epoch': 0.47}
{'loss': 1.1246, 'grad_norm': 1.0240192413330078, 'learning_rate': 2.5e-05, 'epoch': 0.5}
{'loss': 1.4898, 'grad_norm': 0.9574621319770813, 'learning_rate': 2.303852260680388e-05, 'epoch': 0.53}
{'loss': 1.3273, 'grad_norm': 0.9618223309516907, 'learning_rate': 2.1089138373994223e-05, 'epoch': 0.55}
{'loss': 1.3038, 'grad_norm': 0.9145978689193726, 'learning_rate': 1.9163865903602374e-05, 'epoch': 0.57}
{'loss': 1.3025, 'grad_norm': 0.980766773223877, 'learning_rate': 1.7274575140626318e-05, 'epoch': 0.6}
{'eval_loss': 1.3347593545913696, 'eval_runtime': 317.5016, 'eval_samples_per_second': 3.474, 'eval_steps_per_second': 3.474, 'epoch': 0.6} {'loss': 1.3262, 'grad_norm': 2.370988130569458, 'learning_rate': 1.5432914190872757e-05, 'epoch': 0.62}
{'loss': 1.1936, 'grad_norm': 0.9262142181396484, 'learning_rate': 1.3650237506511331e-05, 'epoch': 0.65}
{'loss': 1.3258, 'grad_norm': 1.1432793140411377, 'learning_rate': 1.1937535882101281e-05, 'epoch': 0.68}
{'loss': 1.2491, 'grad_norm': 0.7737239599227905, 'learning_rate': 1.0305368692688174e-05, 'epoch': 0.7}
{'loss': 1.2071, 'grad_norm': 0.917812705039978, 'learning_rate': 8.763798791745411e-06, 'epoch': 0.72}
{'loss': 1.3731, 'grad_norm': 1.0561238527297974, 'learning_rate': 7.3223304703363135e-06, 'epoch': 0.75}
{'loss': 1.3551, 'grad_norm': 1.0381251573562622, 'learning_rate': 5.989850859999227e-06, 'epoch': 0.78}
{'loss': 1.4761, 'grad_norm': 1.004711627960205, 'learning_rate': 4.7745751406263165e-06, 'epoch': 0.8}
{'eval_loss': 1.332088828086853, 'eval_runtime': 317.4536, 'eval_samples_per_second': 3.475, 'eval_steps_per_second': 3.475, 'epoch': 0.8} {'loss': 1.3827, 'grad_norm': 0.9905529022216797, 'learning_rate': 3.6839958911476957e-06, 'epoch': 0.82}
{'loss': 1.5409, 'grad_norm': 0.9593507051467896, 'learning_rate': 2.7248368952908053e-06, 'epoch': 0.85}
{'loss': 1.2229, 'grad_norm': 1.3969199657440186, 'learning_rate': 1.9030116872178316e-06, 'epoch': 0.88}
{'loss': 1.4232, 'grad_norm': 1.2699376344680786, 'learning_rate': 1.2235870926211619e-06, 'epoch': 0.9}
{'loss': 1.3607, 'grad_norm': 1.0963711738586426, 'learning_rate': 6.907519900580861e-07, 'epoch': 0.93}
{'loss': 1.4084, 'grad_norm': 0.961476743221283, 'learning_rate': 3.077914851215585e-07, 'epoch': 0.95}
{'loss': 1.3586, 'grad_norm': 1.0301530361175537, 'learning_rate': 7.706665667180091e-08, 'epoch': 0.97}
{'loss': 1.5, 'grad_norm': 0.9526334404945374, 'learning_rate': 0.0, 'epoch': 1.0}
{'eval_loss': 1.3315749168395996, 'eval_runtime': 317.9966, 'eval_samples_per_second': 3.469, 'eval_steps_per_second': 3.469, 'epoch': 1.0} {'train_runtime': 1929.0292, 'train_samples_per_second': 0.207, 'train_steps_per_second': 0.207, 'train_tokens_per_second': 212.335, 'train_loss': 1.3933543395996093, 'epoch': 1.0} 100%|██████████████████████████████████████████████████████████████████████████████████████| 400/400 [32:09<00:00, 4.82s/it] --- total: 1946.99866771698 s --- evaluation --- eval_problem_0 eval_problem_1 eval_problem_2 eval_problem_3 eval_problem_4 eval_problem_5 eval_problem_6 eval_problem_7 eval_problem_8 eval_problem_9 eval_problem_10 eval_problem_11 eval_problem_12 eval_problem_13 eval_problem_14 eval_problem_15 eval_problem_16 eval_problem_17 eval_problem_18 eval_problem_19 eval_problem_20 eval_problem_21 eval_problem_22 eval_problem_23 eval_problem_24 eval_problem_25 eval_problem_26 eval_problem_27 eval_problem_28 eval_problem_29 eval_problem_30 eval_problem_31 eval_problem_32 eval_problem_33 eval_problem_34 eval_problem_35 eval_problem_36 eval_problem_37 eval_problem_38 eval_problem_39 eval_problem_40 eval_problem_41 eval_problem_42 eval_problem_43 eval_problem_44 eval_problem_45 eval_problem_46 eval_problem_47 eval_problem_48 eval_problem_49 --- total: 2895.8622257709503 s