vit-accident-image / trainer_state.json
tiya1012's picture
cheers
3f5c058
raw
history blame contribute delete
No virus
7.58 kB
{
"best_metric": 0.2074955552816391,
"best_model_checkpoint": "./vit-accident-image/checkpoint-200",
"epoch": 10.0,
"global_step": 500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.2,
"learning_rate": 0.000196,
"loss": 0.6842,
"step": 10
},
{
"epoch": 0.4,
"learning_rate": 0.000192,
"loss": 0.6515,
"step": 20
},
{
"epoch": 0.6,
"learning_rate": 0.000188,
"loss": 0.5938,
"step": 30
},
{
"epoch": 0.8,
"learning_rate": 0.00018400000000000003,
"loss": 0.5845,
"step": 40
},
{
"epoch": 1.0,
"learning_rate": 0.00018,
"loss": 0.5319,
"step": 50
},
{
"epoch": 1.2,
"learning_rate": 0.00017600000000000002,
"loss": 0.2824,
"step": 60
},
{
"epoch": 1.4,
"learning_rate": 0.000172,
"loss": 0.2948,
"step": 70
},
{
"epoch": 1.6,
"learning_rate": 0.000168,
"loss": 0.6173,
"step": 80
},
{
"epoch": 1.8,
"learning_rate": 0.000164,
"loss": 0.2863,
"step": 90
},
{
"epoch": 2.0,
"learning_rate": 0.00016,
"loss": 0.3546,
"step": 100
},
{
"epoch": 2.0,
"eval_accuracy": 0.9183673469387755,
"eval_f1": 0.918435459290099,
"eval_loss": 0.23273873329162598,
"eval_runtime": 2.7024,
"eval_samples_per_second": 36.264,
"eval_steps_per_second": 4.81,
"step": 100
},
{
"epoch": 2.2,
"learning_rate": 0.00015600000000000002,
"loss": 0.1067,
"step": 110
},
{
"epoch": 2.4,
"learning_rate": 0.000152,
"loss": 0.2548,
"step": 120
},
{
"epoch": 2.6,
"learning_rate": 0.000148,
"loss": 0.1812,
"step": 130
},
{
"epoch": 2.8,
"learning_rate": 0.000144,
"loss": 0.2835,
"step": 140
},
{
"epoch": 3.0,
"learning_rate": 0.00014,
"loss": 0.2162,
"step": 150
},
{
"epoch": 3.2,
"learning_rate": 0.00013600000000000003,
"loss": 0.1976,
"step": 160
},
{
"epoch": 3.4,
"learning_rate": 0.000132,
"loss": 0.1068,
"step": 170
},
{
"epoch": 3.6,
"learning_rate": 0.00012800000000000002,
"loss": 0.0842,
"step": 180
},
{
"epoch": 3.8,
"learning_rate": 0.000124,
"loss": 0.168,
"step": 190
},
{
"epoch": 4.0,
"learning_rate": 0.00012,
"loss": 0.1654,
"step": 200
},
{
"epoch": 4.0,
"eval_accuracy": 0.9387755102040817,
"eval_f1": 0.9387755102040817,
"eval_loss": 0.2074955552816391,
"eval_runtime": 2.2462,
"eval_samples_per_second": 43.629,
"eval_steps_per_second": 5.787,
"step": 200
},
{
"epoch": 4.2,
"learning_rate": 0.000116,
"loss": 0.0843,
"step": 210
},
{
"epoch": 4.4,
"learning_rate": 0.00011200000000000001,
"loss": 0.032,
"step": 220
},
{
"epoch": 4.6,
"learning_rate": 0.00010800000000000001,
"loss": 0.1275,
"step": 230
},
{
"epoch": 4.8,
"learning_rate": 0.00010400000000000001,
"loss": 0.073,
"step": 240
},
{
"epoch": 5.0,
"learning_rate": 0.0001,
"loss": 0.1203,
"step": 250
},
{
"epoch": 5.2,
"learning_rate": 9.6e-05,
"loss": 0.0723,
"step": 260
},
{
"epoch": 5.4,
"learning_rate": 9.200000000000001e-05,
"loss": 0.0184,
"step": 270
},
{
"epoch": 5.6,
"learning_rate": 8.800000000000001e-05,
"loss": 0.0633,
"step": 280
},
{
"epoch": 5.8,
"learning_rate": 8.4e-05,
"loss": 0.0989,
"step": 290
},
{
"epoch": 6.0,
"learning_rate": 8e-05,
"loss": 0.0146,
"step": 300
},
{
"epoch": 6.0,
"eval_accuracy": 0.9387755102040817,
"eval_f1": 0.9386728276216146,
"eval_loss": 0.24968579411506653,
"eval_runtime": 2.3881,
"eval_samples_per_second": 41.036,
"eval_steps_per_second": 5.444,
"step": 300
},
{
"epoch": 6.2,
"learning_rate": 7.6e-05,
"loss": 0.0295,
"step": 310
},
{
"epoch": 6.4,
"learning_rate": 7.2e-05,
"loss": 0.0698,
"step": 320
},
{
"epoch": 6.6,
"learning_rate": 6.800000000000001e-05,
"loss": 0.0289,
"step": 330
},
{
"epoch": 6.8,
"learning_rate": 6.400000000000001e-05,
"loss": 0.0336,
"step": 340
},
{
"epoch": 7.0,
"learning_rate": 6e-05,
"loss": 0.0384,
"step": 350
},
{
"epoch": 7.2,
"learning_rate": 5.6000000000000006e-05,
"loss": 0.0344,
"step": 360
},
{
"epoch": 7.4,
"learning_rate": 5.2000000000000004e-05,
"loss": 0.0085,
"step": 370
},
{
"epoch": 7.6,
"learning_rate": 4.8e-05,
"loss": 0.0145,
"step": 380
},
{
"epoch": 7.8,
"learning_rate": 4.4000000000000006e-05,
"loss": 0.0064,
"step": 390
},
{
"epoch": 8.0,
"learning_rate": 4e-05,
"loss": 0.0317,
"step": 400
},
{
"epoch": 8.0,
"eval_accuracy": 0.9285714285714286,
"eval_f1": 0.9285190999476715,
"eval_loss": 0.21786272525787354,
"eval_runtime": 3.0699,
"eval_samples_per_second": 31.922,
"eval_steps_per_second": 4.235,
"step": 400
},
{
"epoch": 8.2,
"learning_rate": 3.6e-05,
"loss": 0.0081,
"step": 410
},
{
"epoch": 8.4,
"learning_rate": 3.2000000000000005e-05,
"loss": 0.0057,
"step": 420
},
{
"epoch": 8.6,
"learning_rate": 2.8000000000000003e-05,
"loss": 0.0364,
"step": 430
},
{
"epoch": 8.8,
"learning_rate": 2.4e-05,
"loss": 0.0062,
"step": 440
},
{
"epoch": 9.0,
"learning_rate": 2e-05,
"loss": 0.025,
"step": 450
},
{
"epoch": 9.2,
"learning_rate": 1.6000000000000003e-05,
"loss": 0.0087,
"step": 460
},
{
"epoch": 9.4,
"learning_rate": 1.2e-05,
"loss": 0.0136,
"step": 470
},
{
"epoch": 9.6,
"learning_rate": 8.000000000000001e-06,
"loss": 0.0228,
"step": 480
},
{
"epoch": 9.8,
"learning_rate": 4.000000000000001e-06,
"loss": 0.005,
"step": 490
},
{
"epoch": 10.0,
"learning_rate": 0.0,
"loss": 0.0192,
"step": 500
},
{
"epoch": 10.0,
"eval_accuracy": 0.9285714285714286,
"eval_f1": 0.9286087125109987,
"eval_loss": 0.2254808396100998,
"eval_runtime": 2.2207,
"eval_samples_per_second": 44.131,
"eval_steps_per_second": 5.854,
"step": 500
},
{
"epoch": 10.0,
"step": 500,
"total_flos": 6.129616378514227e+17,
"train_loss": 0.15593594065308572,
"train_runtime": 520.5132,
"train_samples_per_second": 15.197,
"train_steps_per_second": 0.961
}
],
"max_steps": 500,
"num_train_epochs": 10,
"total_flos": 6.129616378514227e+17,
"trial_name": null,
"trial_params": null
}