sst2_bert-base-uncased_81 / trainer_state.json
henryscheible's picture
End of training
be692ad
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 5.0,
"global_step": 10525,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.24,
"learning_rate": 1.9049881235154395e-05,
"loss": 0.3458,
"step": 500
},
{
"epoch": 0.48,
"learning_rate": 1.809976247030879e-05,
"loss": 0.2336,
"step": 1000
},
{
"epoch": 0.71,
"learning_rate": 1.7149643705463184e-05,
"loss": 0.1974,
"step": 1500
},
{
"epoch": 0.95,
"learning_rate": 1.6199524940617578e-05,
"loss": 0.1818,
"step": 2000
},
{
"epoch": 1.19,
"learning_rate": 1.5249406175771972e-05,
"loss": 0.1369,
"step": 2500
},
{
"epoch": 1.43,
"learning_rate": 1.4299287410926367e-05,
"loss": 0.1215,
"step": 3000
},
{
"epoch": 1.66,
"learning_rate": 1.3349168646080762e-05,
"loss": 0.1225,
"step": 3500
},
{
"epoch": 1.9,
"learning_rate": 1.2399049881235155e-05,
"loss": 0.1279,
"step": 4000
},
{
"epoch": 2.14,
"learning_rate": 1.144893111638955e-05,
"loss": 0.0959,
"step": 4500
},
{
"epoch": 2.38,
"learning_rate": 1.0498812351543943e-05,
"loss": 0.079,
"step": 5000
},
{
"epoch": 2.61,
"learning_rate": 9.548693586698338e-06,
"loss": 0.089,
"step": 5500
},
{
"epoch": 2.85,
"learning_rate": 8.598574821852733e-06,
"loss": 0.089,
"step": 6000
},
{
"epoch": 3.09,
"learning_rate": 7.648456057007126e-06,
"loss": 0.0773,
"step": 6500
},
{
"epoch": 3.33,
"learning_rate": 6.698337292161521e-06,
"loss": 0.0641,
"step": 7000
},
{
"epoch": 3.56,
"learning_rate": 5.748218527315916e-06,
"loss": 0.063,
"step": 7500
},
{
"epoch": 3.8,
"learning_rate": 4.798099762470309e-06,
"loss": 0.0638,
"step": 8000
},
{
"epoch": 4.04,
"learning_rate": 3.8479809976247036e-06,
"loss": 0.0571,
"step": 8500
},
{
"epoch": 4.28,
"learning_rate": 2.897862232779098e-06,
"loss": 0.0434,
"step": 9000
},
{
"epoch": 4.51,
"learning_rate": 1.947743467933492e-06,
"loss": 0.0416,
"step": 9500
},
{
"epoch": 4.75,
"learning_rate": 9.976247030878861e-07,
"loss": 0.0469,
"step": 10000
},
{
"epoch": 4.99,
"learning_rate": 4.750593824228029e-08,
"loss": 0.0443,
"step": 10500
},
{
"epoch": 5.0,
"step": 10525,
"total_flos": 0.0,
"train_loss": 0.11035192067719413,
"train_runtime": 4555.7313,
"train_samples_per_second": 73.917,
"train_steps_per_second": 2.31
}
],
"max_steps": 10525,
"num_train_epochs": 5,
"total_flos": 0.0,
"trial_name": null,
"trial_params": null
}