lora-hellaswag / trainer_state.json
igzi's picture
Upload latest checkpoint for hellaswag
72c2965 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.08333333333333333,
"eval_steps": 500,
"global_step": 5000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0008333333333333334,
"grad_norm": 0.1435546875,
"learning_rate": 0.00019800000000000002,
"loss": 0.1418,
"step": 50
},
{
"epoch": 0.0016666666666666668,
"grad_norm": 0.0118408203125,
"learning_rate": 0.000196,
"loss": 0.1015,
"step": 100
},
{
"epoch": 0.0025,
"grad_norm": 0.059326171875,
"learning_rate": 0.000194,
"loss": 0.0625,
"step": 150
},
{
"epoch": 0.0033333333333333335,
"grad_norm": 0.00128936767578125,
"learning_rate": 0.000192,
"loss": 0.0441,
"step": 200
},
{
"epoch": 0.004166666666666667,
"grad_norm": 0.02783203125,
"learning_rate": 0.00019,
"loss": 0.0547,
"step": 250
},
{
"epoch": 0.005,
"grad_norm": 12.625,
"learning_rate": 0.000188,
"loss": 0.074,
"step": 300
},
{
"epoch": 0.005833333333333334,
"grad_norm": 0.09765625,
"learning_rate": 0.00018600000000000002,
"loss": 0.0284,
"step": 350
},
{
"epoch": 0.006666666666666667,
"grad_norm": 0.130859375,
"learning_rate": 0.00018400000000000003,
"loss": 0.0453,
"step": 400
},
{
"epoch": 0.0075,
"grad_norm": 9.625,
"learning_rate": 0.000182,
"loss": 0.054,
"step": 450
},
{
"epoch": 0.008333333333333333,
"grad_norm": 0.0006103515625,
"learning_rate": 0.00018,
"loss": 0.0222,
"step": 500
},
{
"epoch": 0.009166666666666667,
"grad_norm": 4.65625,
"learning_rate": 0.00017800000000000002,
"loss": 0.0493,
"step": 550
},
{
"epoch": 0.01,
"grad_norm": 0.06201171875,
"learning_rate": 0.00017600000000000002,
"loss": 0.0522,
"step": 600
},
{
"epoch": 0.010833333333333334,
"grad_norm": 0.00885009765625,
"learning_rate": 0.000174,
"loss": 0.0526,
"step": 650
},
{
"epoch": 0.011666666666666667,
"grad_norm": 7.53125,
"learning_rate": 0.000172,
"loss": 0.0393,
"step": 700
},
{
"epoch": 0.0125,
"grad_norm": 0.00579833984375,
"learning_rate": 0.00017,
"loss": 0.0678,
"step": 750
},
{
"epoch": 0.013333333333333334,
"grad_norm": 0.0031890869140625,
"learning_rate": 0.000168,
"loss": 0.0415,
"step": 800
},
{
"epoch": 0.014166666666666666,
"grad_norm": 0.00010061264038085938,
"learning_rate": 0.000166,
"loss": 0.0202,
"step": 850
},
{
"epoch": 0.015,
"grad_norm": 1.09375,
"learning_rate": 0.000164,
"loss": 0.0695,
"step": 900
},
{
"epoch": 0.015833333333333335,
"grad_norm": 0.65625,
"learning_rate": 0.000162,
"loss": 0.0507,
"step": 950
},
{
"epoch": 0.016666666666666666,
"grad_norm": 0.00701904296875,
"learning_rate": 0.00016,
"loss": 0.0201,
"step": 1000
},
{
"epoch": 0.0175,
"grad_norm": 9.3125,
"learning_rate": 0.00015800000000000002,
"loss": 0.0593,
"step": 1050
},
{
"epoch": 0.018333333333333333,
"grad_norm": 7.46875,
"learning_rate": 0.00015600000000000002,
"loss": 0.0149,
"step": 1100
},
{
"epoch": 0.019166666666666665,
"grad_norm": 0.004058837890625,
"learning_rate": 0.000154,
"loss": 0.0344,
"step": 1150
},
{
"epoch": 0.02,
"grad_norm": 15.3125,
"learning_rate": 0.000152,
"loss": 0.054,
"step": 1200
},
{
"epoch": 0.020833333333333332,
"grad_norm": 18.125,
"learning_rate": 0.00015000000000000001,
"loss": 0.023,
"step": 1250
},
{
"epoch": 0.021666666666666667,
"grad_norm": 0.0771484375,
"learning_rate": 0.000148,
"loss": 0.0474,
"step": 1300
},
{
"epoch": 0.0225,
"grad_norm": 0.005615234375,
"learning_rate": 0.000146,
"loss": 0.0042,
"step": 1350
},
{
"epoch": 0.023333333333333334,
"grad_norm": 8.8125,
"learning_rate": 0.000144,
"loss": 0.0477,
"step": 1400
},
{
"epoch": 0.024166666666666666,
"grad_norm": 0.006988525390625,
"learning_rate": 0.000142,
"loss": 0.0268,
"step": 1450
},
{
"epoch": 0.025,
"grad_norm": 6.0625,
"learning_rate": 0.00014,
"loss": 0.0224,
"step": 1500
},
{
"epoch": 0.025833333333333333,
"grad_norm": 4.03125,
"learning_rate": 0.000138,
"loss": 0.0369,
"step": 1550
},
{
"epoch": 0.02666666666666667,
"grad_norm": 0.59375,
"learning_rate": 0.00013600000000000003,
"loss": 0.01,
"step": 1600
},
{
"epoch": 0.0275,
"grad_norm": 0.004974365234375,
"learning_rate": 0.000134,
"loss": 0.0621,
"step": 1650
},
{
"epoch": 0.028333333333333332,
"grad_norm": 3.75,
"learning_rate": 0.000132,
"loss": 0.0915,
"step": 1700
},
{
"epoch": 0.029166666666666667,
"grad_norm": 0.095703125,
"learning_rate": 0.00013000000000000002,
"loss": 0.0215,
"step": 1750
},
{
"epoch": 0.03,
"grad_norm": 5.625,
"learning_rate": 0.00012800000000000002,
"loss": 0.034,
"step": 1800
},
{
"epoch": 0.030833333333333334,
"grad_norm": 0.000701904296875,
"learning_rate": 0.000126,
"loss": 0.028,
"step": 1850
},
{
"epoch": 0.03166666666666667,
"grad_norm": 11.9375,
"learning_rate": 0.000124,
"loss": 0.0136,
"step": 1900
},
{
"epoch": 0.0325,
"grad_norm": 0.0030517578125,
"learning_rate": 0.000122,
"loss": 0.0305,
"step": 1950
},
{
"epoch": 0.03333333333333333,
"grad_norm": 0.000507354736328125,
"learning_rate": 0.00012,
"loss": 0.0254,
"step": 2000
},
{
"epoch": 0.034166666666666665,
"grad_norm": 0.00872802734375,
"learning_rate": 0.000118,
"loss": 0.0291,
"step": 2050
},
{
"epoch": 0.035,
"grad_norm": 0.03076171875,
"learning_rate": 0.000116,
"loss": 0.0422,
"step": 2100
},
{
"epoch": 0.035833333333333335,
"grad_norm": 0.000885009765625,
"learning_rate": 0.00011399999999999999,
"loss": 0.0175,
"step": 2150
},
{
"epoch": 0.03666666666666667,
"grad_norm": 7.625,
"learning_rate": 0.00011200000000000001,
"loss": 0.0476,
"step": 2200
},
{
"epoch": 0.0375,
"grad_norm": 0.0013885498046875,
"learning_rate": 0.00011000000000000002,
"loss": 0.036,
"step": 2250
},
{
"epoch": 0.03833333333333333,
"grad_norm": 0.040771484375,
"learning_rate": 0.00010800000000000001,
"loss": 0.0182,
"step": 2300
},
{
"epoch": 0.03916666666666667,
"grad_norm": 0.3203125,
"learning_rate": 0.00010600000000000002,
"loss": 0.0207,
"step": 2350
},
{
"epoch": 0.04,
"grad_norm": 0.022216796875,
"learning_rate": 0.00010400000000000001,
"loss": 0.005,
"step": 2400
},
{
"epoch": 0.04083333333333333,
"grad_norm": 0.091796875,
"learning_rate": 0.00010200000000000001,
"loss": 0.009,
"step": 2450
},
{
"epoch": 0.041666666666666664,
"grad_norm": 0.000171661376953125,
"learning_rate": 0.0001,
"loss": 0.006,
"step": 2500
},
{
"epoch": 0.0425,
"grad_norm": 0.052001953125,
"learning_rate": 9.8e-05,
"loss": 0.0407,
"step": 2550
},
{
"epoch": 0.043333333333333335,
"grad_norm": 7.03125,
"learning_rate": 9.6e-05,
"loss": 0.0313,
"step": 2600
},
{
"epoch": 0.04416666666666667,
"grad_norm": 0.0361328125,
"learning_rate": 9.4e-05,
"loss": 0.0196,
"step": 2650
},
{
"epoch": 0.045,
"grad_norm": 0.205078125,
"learning_rate": 9.200000000000001e-05,
"loss": 0.0303,
"step": 2700
},
{
"epoch": 0.04583333333333333,
"grad_norm": 0.00360107421875,
"learning_rate": 9e-05,
"loss": 0.0294,
"step": 2750
},
{
"epoch": 0.04666666666666667,
"grad_norm": 0.00191497802734375,
"learning_rate": 8.800000000000001e-05,
"loss": 0.02,
"step": 2800
},
{
"epoch": 0.0475,
"grad_norm": 0.10693359375,
"learning_rate": 8.6e-05,
"loss": 0.02,
"step": 2850
},
{
"epoch": 0.04833333333333333,
"grad_norm": 0.0028076171875,
"learning_rate": 8.4e-05,
"loss": 0.0562,
"step": 2900
},
{
"epoch": 0.049166666666666664,
"grad_norm": 5.03125,
"learning_rate": 8.2e-05,
"loss": 0.0047,
"step": 2950
},
{
"epoch": 0.05,
"grad_norm": 0.000835418701171875,
"learning_rate": 8e-05,
"loss": 0.0375,
"step": 3000
},
{
"epoch": 0.050833333333333335,
"grad_norm": 0.0021820068359375,
"learning_rate": 7.800000000000001e-05,
"loss": 0.0232,
"step": 3050
},
{
"epoch": 0.051666666666666666,
"grad_norm": 0.009765625,
"learning_rate": 7.6e-05,
"loss": 0.0472,
"step": 3100
},
{
"epoch": 0.0525,
"grad_norm": 0.00323486328125,
"learning_rate": 7.4e-05,
"loss": 0.0294,
"step": 3150
},
{
"epoch": 0.05333333333333334,
"grad_norm": 0.005767822265625,
"learning_rate": 7.2e-05,
"loss": 0.0233,
"step": 3200
},
{
"epoch": 0.05416666666666667,
"grad_norm": 0.01165771484375,
"learning_rate": 7e-05,
"loss": 0.0216,
"step": 3250
},
{
"epoch": 0.055,
"grad_norm": 0.01806640625,
"learning_rate": 6.800000000000001e-05,
"loss": 0.0076,
"step": 3300
},
{
"epoch": 0.05583333333333333,
"grad_norm": 0.0019989013671875,
"learning_rate": 6.6e-05,
"loss": 0.0058,
"step": 3350
},
{
"epoch": 0.056666666666666664,
"grad_norm": 0.00091552734375,
"learning_rate": 6.400000000000001e-05,
"loss": 0.0171,
"step": 3400
},
{
"epoch": 0.0575,
"grad_norm": 0.392578125,
"learning_rate": 6.2e-05,
"loss": 0.0408,
"step": 3450
},
{
"epoch": 0.058333333333333334,
"grad_norm": 0.0006103515625,
"learning_rate": 6e-05,
"loss": 0.0071,
"step": 3500
},
{
"epoch": 0.059166666666666666,
"grad_norm": 0.0004730224609375,
"learning_rate": 5.8e-05,
"loss": 0.0134,
"step": 3550
},
{
"epoch": 0.06,
"grad_norm": 0.007476806640625,
"learning_rate": 5.6000000000000006e-05,
"loss": 0.0507,
"step": 3600
},
{
"epoch": 0.060833333333333336,
"grad_norm": 0.2177734375,
"learning_rate": 5.4000000000000005e-05,
"loss": 0.043,
"step": 3650
},
{
"epoch": 0.06166666666666667,
"grad_norm": 4.875,
"learning_rate": 5.2000000000000004e-05,
"loss": 0.0414,
"step": 3700
},
{
"epoch": 0.0625,
"grad_norm": 0.2255859375,
"learning_rate": 5e-05,
"loss": 0.0337,
"step": 3750
},
{
"epoch": 0.06333333333333334,
"grad_norm": 1.484375,
"learning_rate": 4.8e-05,
"loss": 0.013,
"step": 3800
},
{
"epoch": 0.06416666666666666,
"grad_norm": 0.05322265625,
"learning_rate": 4.600000000000001e-05,
"loss": 0.0137,
"step": 3850
},
{
"epoch": 0.065,
"grad_norm": 0.0023651123046875,
"learning_rate": 4.4000000000000006e-05,
"loss": 0.0104,
"step": 3900
},
{
"epoch": 0.06583333333333333,
"grad_norm": 4.8125,
"learning_rate": 4.2e-05,
"loss": 0.0191,
"step": 3950
},
{
"epoch": 0.06666666666666667,
"grad_norm": 0.1318359375,
"learning_rate": 4e-05,
"loss": 0.0056,
"step": 4000
},
{
"epoch": 0.0675,
"grad_norm": 0.02978515625,
"learning_rate": 3.8e-05,
"loss": 0.0446,
"step": 4050
},
{
"epoch": 0.06833333333333333,
"grad_norm": 0.2138671875,
"learning_rate": 3.6e-05,
"loss": 0.0145,
"step": 4100
},
{
"epoch": 0.06916666666666667,
"grad_norm": 0.0269775390625,
"learning_rate": 3.4000000000000007e-05,
"loss": 0.0248,
"step": 4150
},
{
"epoch": 0.07,
"grad_norm": 0.0264892578125,
"learning_rate": 3.2000000000000005e-05,
"loss": 0.0452,
"step": 4200
},
{
"epoch": 0.07083333333333333,
"grad_norm": 0.005889892578125,
"learning_rate": 3e-05,
"loss": 0.017,
"step": 4250
},
{
"epoch": 0.07166666666666667,
"grad_norm": 0.009033203125,
"learning_rate": 2.8000000000000003e-05,
"loss": 0.0209,
"step": 4300
},
{
"epoch": 0.0725,
"grad_norm": 0.0277099609375,
"learning_rate": 2.6000000000000002e-05,
"loss": 0.0159,
"step": 4350
},
{
"epoch": 0.07333333333333333,
"grad_norm": 0.0036468505859375,
"learning_rate": 2.4e-05,
"loss": 0.0167,
"step": 4400
},
{
"epoch": 0.07416666666666667,
"grad_norm": 0.006866455078125,
"learning_rate": 2.2000000000000003e-05,
"loss": 0.0353,
"step": 4450
},
{
"epoch": 0.075,
"grad_norm": 0.00173187255859375,
"learning_rate": 2e-05,
"loss": 0.0319,
"step": 4500
},
{
"epoch": 0.07583333333333334,
"grad_norm": 0.005096435546875,
"learning_rate": 1.8e-05,
"loss": 0.0131,
"step": 4550
},
{
"epoch": 0.07666666666666666,
"grad_norm": 1.6328125,
"learning_rate": 1.6000000000000003e-05,
"loss": 0.0371,
"step": 4600
},
{
"epoch": 0.0775,
"grad_norm": 0.00139617919921875,
"learning_rate": 1.4000000000000001e-05,
"loss": 0.0254,
"step": 4650
},
{
"epoch": 0.07833333333333334,
"grad_norm": 0.01397705078125,
"learning_rate": 1.2e-05,
"loss": 0.031,
"step": 4700
},
{
"epoch": 0.07916666666666666,
"grad_norm": 0.255859375,
"learning_rate": 1e-05,
"loss": 0.0206,
"step": 4750
},
{
"epoch": 0.08,
"grad_norm": 0.0252685546875,
"learning_rate": 8.000000000000001e-06,
"loss": 0.0232,
"step": 4800
},
{
"epoch": 0.08083333333333333,
"grad_norm": 0.000843048095703125,
"learning_rate": 6e-06,
"loss": 0.0104,
"step": 4850
},
{
"epoch": 0.08166666666666667,
"grad_norm": 0.029296875,
"learning_rate": 4.000000000000001e-06,
"loss": 0.0055,
"step": 4900
},
{
"epoch": 0.0825,
"grad_norm": 0.0147705078125,
"learning_rate": 2.0000000000000003e-06,
"loss": 0.0453,
"step": 4950
},
{
"epoch": 0.08333333333333333,
"grad_norm": 0.00762939453125,
"learning_rate": 0.0,
"loss": 0.0089,
"step": 5000
}
],
"logging_steps": 50,
"max_steps": 5000,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 5.492272922714112e+16,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}