swiftformer-xs-dmae-va-U-40 / trainer_state.json
Augusto777's picture
End of training
cf7506a verified
raw
history blame
No virus
13.1 kB
{
"best_metric": 0.7522935779816514,
"best_model_checkpoint": "swiftformer-xs-dmae-va-U-40/checkpoint-224",
"epoch": 36.12903225806452,
"eval_steps": 500,
"global_step": 280,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.9,
"eval_accuracy": 0.29357798165137616,
"eval_loss": 1.3578405380249023,
"eval_runtime": 0.4941,
"eval_samples_per_second": 220.605,
"eval_steps_per_second": 8.096,
"step": 7
},
{
"epoch": 1.29,
"learning_rate": 1.785714285714286e-05,
"loss": 1.3702,
"step": 10
},
{
"epoch": 1.94,
"eval_accuracy": 0.29357798165137616,
"eval_loss": 1.3703261613845825,
"eval_runtime": 0.5833,
"eval_samples_per_second": 186.883,
"eval_steps_per_second": 6.858,
"step": 15
},
{
"epoch": 2.58,
"learning_rate": 3.571428571428572e-05,
"loss": 1.3497,
"step": 20
},
{
"epoch": 2.97,
"eval_accuracy": 0.3394495412844037,
"eval_loss": 1.3361120223999023,
"eval_runtime": 0.4226,
"eval_samples_per_second": 257.899,
"eval_steps_per_second": 9.464,
"step": 23
},
{
"epoch": 3.87,
"learning_rate": 4.960317460317461e-05,
"loss": 1.3004,
"step": 30
},
{
"epoch": 4.0,
"eval_accuracy": 0.3669724770642202,
"eval_loss": 1.2852015495300293,
"eval_runtime": 0.4154,
"eval_samples_per_second": 262.426,
"eval_steps_per_second": 9.63,
"step": 31
},
{
"epoch": 4.9,
"eval_accuracy": 0.43119266055045874,
"eval_loss": 1.231681227684021,
"eval_runtime": 0.4157,
"eval_samples_per_second": 262.204,
"eval_steps_per_second": 9.622,
"step": 38
},
{
"epoch": 5.16,
"learning_rate": 4.761904761904762e-05,
"loss": 1.2248,
"step": 40
},
{
"epoch": 5.94,
"eval_accuracy": 0.45871559633027525,
"eval_loss": 1.1785550117492676,
"eval_runtime": 0.4211,
"eval_samples_per_second": 258.867,
"eval_steps_per_second": 9.5,
"step": 46
},
{
"epoch": 6.45,
"learning_rate": 4.563492063492064e-05,
"loss": 1.1485,
"step": 50
},
{
"epoch": 6.97,
"eval_accuracy": 0.5045871559633027,
"eval_loss": 1.123984694480896,
"eval_runtime": 0.4143,
"eval_samples_per_second": 263.124,
"eval_steps_per_second": 9.656,
"step": 54
},
{
"epoch": 7.74,
"learning_rate": 4.3650793650793655e-05,
"loss": 1.0759,
"step": 60
},
{
"epoch": 8.0,
"eval_accuracy": 0.5504587155963303,
"eval_loss": 1.0727368593215942,
"eval_runtime": 0.4109,
"eval_samples_per_second": 265.292,
"eval_steps_per_second": 9.735,
"step": 62
},
{
"epoch": 8.9,
"eval_accuracy": 0.5596330275229358,
"eval_loss": 1.0403714179992676,
"eval_runtime": 0.5431,
"eval_samples_per_second": 200.69,
"eval_steps_per_second": 7.365,
"step": 69
},
{
"epoch": 9.03,
"learning_rate": 4.166666666666667e-05,
"loss": 1.0244,
"step": 70
},
{
"epoch": 9.94,
"eval_accuracy": 0.6238532110091743,
"eval_loss": 0.974201500415802,
"eval_runtime": 0.4104,
"eval_samples_per_second": 265.623,
"eval_steps_per_second": 9.748,
"step": 77
},
{
"epoch": 10.32,
"learning_rate": 3.968253968253968e-05,
"loss": 0.9782,
"step": 80
},
{
"epoch": 10.97,
"eval_accuracy": 0.6422018348623854,
"eval_loss": 0.9374117851257324,
"eval_runtime": 0.6713,
"eval_samples_per_second": 162.361,
"eval_steps_per_second": 5.958,
"step": 85
},
{
"epoch": 11.61,
"learning_rate": 3.76984126984127e-05,
"loss": 0.9359,
"step": 90
},
{
"epoch": 12.0,
"eval_accuracy": 0.6788990825688074,
"eval_loss": 0.9196614027023315,
"eval_runtime": 0.4175,
"eval_samples_per_second": 261.061,
"eval_steps_per_second": 9.58,
"step": 93
},
{
"epoch": 12.9,
"learning_rate": 3.571428571428572e-05,
"loss": 0.9051,
"step": 100
},
{
"epoch": 12.9,
"eval_accuracy": 0.6880733944954128,
"eval_loss": 0.8753331899642944,
"eval_runtime": 0.41,
"eval_samples_per_second": 265.867,
"eval_steps_per_second": 9.757,
"step": 100
},
{
"epoch": 13.94,
"eval_accuracy": 0.6972477064220184,
"eval_loss": 0.8679403066635132,
"eval_runtime": 0.4077,
"eval_samples_per_second": 267.322,
"eval_steps_per_second": 9.81,
"step": 108
},
{
"epoch": 14.19,
"learning_rate": 3.3730158730158734e-05,
"loss": 0.8652,
"step": 110
},
{
"epoch": 14.97,
"eval_accuracy": 0.7155963302752294,
"eval_loss": 0.8316473364830017,
"eval_runtime": 0.4256,
"eval_samples_per_second": 256.117,
"eval_steps_per_second": 9.399,
"step": 116
},
{
"epoch": 15.48,
"learning_rate": 3.1746031746031745e-05,
"loss": 0.8336,
"step": 120
},
{
"epoch": 16.0,
"eval_accuracy": 0.6972477064220184,
"eval_loss": 0.8222222924232483,
"eval_runtime": 0.4301,
"eval_samples_per_second": 253.434,
"eval_steps_per_second": 9.3,
"step": 124
},
{
"epoch": 16.77,
"learning_rate": 2.9761904761904762e-05,
"loss": 0.8177,
"step": 130
},
{
"epoch": 16.9,
"eval_accuracy": 0.6972477064220184,
"eval_loss": 0.8177938461303711,
"eval_runtime": 0.4098,
"eval_samples_per_second": 266.011,
"eval_steps_per_second": 9.762,
"step": 131
},
{
"epoch": 17.94,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.7817714810371399,
"eval_runtime": 0.5687,
"eval_samples_per_second": 191.682,
"eval_steps_per_second": 7.034,
"step": 139
},
{
"epoch": 18.06,
"learning_rate": 2.777777777777778e-05,
"loss": 0.8077,
"step": 140
},
{
"epoch": 18.97,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.7627159357070923,
"eval_runtime": 0.4134,
"eval_samples_per_second": 263.676,
"eval_steps_per_second": 9.676,
"step": 147
},
{
"epoch": 19.35,
"learning_rate": 2.5793650793650796e-05,
"loss": 0.7796,
"step": 150
},
{
"epoch": 20.0,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.7478492259979248,
"eval_runtime": 0.4144,
"eval_samples_per_second": 263.034,
"eval_steps_per_second": 9.653,
"step": 155
},
{
"epoch": 20.65,
"learning_rate": 2.380952380952381e-05,
"loss": 0.7673,
"step": 160
},
{
"epoch": 20.9,
"eval_accuracy": 0.7431192660550459,
"eval_loss": 0.7414626479148865,
"eval_runtime": 0.408,
"eval_samples_per_second": 267.182,
"eval_steps_per_second": 9.805,
"step": 162
},
{
"epoch": 21.94,
"learning_rate": 2.1825396825396827e-05,
"loss": 0.7445,
"step": 170
},
{
"epoch": 21.94,
"eval_accuracy": 0.7155963302752294,
"eval_loss": 0.7413556575775146,
"eval_runtime": 0.4043,
"eval_samples_per_second": 269.576,
"eval_steps_per_second": 9.893,
"step": 170
},
{
"epoch": 22.97,
"eval_accuracy": 0.7155963302752294,
"eval_loss": 0.7375438809394836,
"eval_runtime": 0.4062,
"eval_samples_per_second": 268.371,
"eval_steps_per_second": 9.848,
"step": 178
},
{
"epoch": 23.23,
"learning_rate": 1.984126984126984e-05,
"loss": 0.7413,
"step": 180
},
{
"epoch": 24.0,
"eval_accuracy": 0.7155963302752294,
"eval_loss": 0.7353999018669128,
"eval_runtime": 0.4171,
"eval_samples_per_second": 261.311,
"eval_steps_per_second": 9.589,
"step": 186
},
{
"epoch": 24.52,
"learning_rate": 1.785714285714286e-05,
"loss": 0.739,
"step": 190
},
{
"epoch": 24.9,
"eval_accuracy": 0.7431192660550459,
"eval_loss": 0.71100252866745,
"eval_runtime": 0.4123,
"eval_samples_per_second": 264.367,
"eval_steps_per_second": 9.702,
"step": 193
},
{
"epoch": 25.81,
"learning_rate": 1.5873015873015872e-05,
"loss": 0.6992,
"step": 200
},
{
"epoch": 25.94,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.7120506763458252,
"eval_runtime": 0.4281,
"eval_samples_per_second": 254.6,
"eval_steps_per_second": 9.343,
"step": 201
},
{
"epoch": 26.97,
"eval_accuracy": 0.7431192660550459,
"eval_loss": 0.7044178247451782,
"eval_runtime": 0.5666,
"eval_samples_per_second": 192.368,
"eval_steps_per_second": 7.059,
"step": 209
},
{
"epoch": 27.1,
"learning_rate": 1.388888888888889e-05,
"loss": 0.7111,
"step": 210
},
{
"epoch": 28.0,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.6947001218795776,
"eval_runtime": 0.4105,
"eval_samples_per_second": 265.532,
"eval_steps_per_second": 9.744,
"step": 217
},
{
"epoch": 28.39,
"learning_rate": 1.1904761904761905e-05,
"loss": 0.7013,
"step": 220
},
{
"epoch": 28.9,
"eval_accuracy": 0.7522935779816514,
"eval_loss": 0.7007379531860352,
"eval_runtime": 0.4106,
"eval_samples_per_second": 265.441,
"eval_steps_per_second": 9.741,
"step": 224
},
{
"epoch": 29.68,
"learning_rate": 9.92063492063492e-06,
"loss": 0.712,
"step": 230
},
{
"epoch": 29.94,
"eval_accuracy": 0.7431192660550459,
"eval_loss": 0.6792589426040649,
"eval_runtime": 0.4139,
"eval_samples_per_second": 263.354,
"eval_steps_per_second": 9.664,
"step": 232
},
{
"epoch": 30.97,
"learning_rate": 7.936507936507936e-06,
"loss": 0.671,
"step": 240
},
{
"epoch": 30.97,
"eval_accuracy": 0.7431192660550459,
"eval_loss": 0.6808269619941711,
"eval_runtime": 0.408,
"eval_samples_per_second": 267.189,
"eval_steps_per_second": 9.805,
"step": 240
},
{
"epoch": 32.0,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.6820599436759949,
"eval_runtime": 0.4244,
"eval_samples_per_second": 256.807,
"eval_steps_per_second": 9.424,
"step": 248
},
{
"epoch": 32.26,
"learning_rate": 5.9523809523809525e-06,
"loss": 0.6862,
"step": 250
},
{
"epoch": 32.9,
"eval_accuracy": 0.7339449541284404,
"eval_loss": 0.6705361008644104,
"eval_runtime": 0.4159,
"eval_samples_per_second": 262.074,
"eval_steps_per_second": 9.617,
"step": 255
},
{
"epoch": 33.55,
"learning_rate": 3.968253968253968e-06,
"loss": 0.6606,
"step": 260
},
{
"epoch": 33.94,
"eval_accuracy": 0.7431192660550459,
"eval_loss": 0.6783888339996338,
"eval_runtime": 0.4197,
"eval_samples_per_second": 259.739,
"eval_steps_per_second": 9.532,
"step": 263
},
{
"epoch": 34.84,
"learning_rate": 1.984126984126984e-06,
"loss": 0.6667,
"step": 270
},
{
"epoch": 34.97,
"eval_accuracy": 0.7522935779816514,
"eval_loss": 0.6764441132545471,
"eval_runtime": 0.4166,
"eval_samples_per_second": 261.642,
"eval_steps_per_second": 9.602,
"step": 271
},
{
"epoch": 36.0,
"eval_accuracy": 0.7522935779816514,
"eval_loss": 0.6716886758804321,
"eval_runtime": 0.5848,
"eval_samples_per_second": 186.395,
"eval_steps_per_second": 6.84,
"step": 279
},
{
"epoch": 36.13,
"learning_rate": 0.0,
"loss": 0.6687,
"step": 280
},
{
"epoch": 36.13,
"eval_accuracy": 0.7522935779816514,
"eval_loss": 0.6736045479774475,
"eval_runtime": 0.4181,
"eval_samples_per_second": 260.729,
"eval_steps_per_second": 9.568,
"step": 280
},
{
"epoch": 36.13,
"step": 280,
"total_flos": 9.686412043576934e+16,
"train_loss": 0.878055340903146,
"train_runtime": 251.6613,
"train_samples_per_second": 155.447,
"train_steps_per_second": 1.113
}
],
"logging_steps": 10,
"max_steps": 280,
"num_train_epochs": 40,
"save_steps": 500,
"total_flos": 9.686412043576934e+16,
"trial_name": null,
"trial_params": null
}