batterybert-cased-squad2 / trainer_state.json
ti250's picture
Upload 15 files
79701c1 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 10.0,
"global_step": 5440,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.92,
"learning_rate": 2.724264705882353e-05,
"loss": 1.6081,
"step": 500
},
{
"epoch": 1.84,
"learning_rate": 2.448529411764706e-05,
"loss": 0.9985,
"step": 1000
},
{
"epoch": 2.76,
"learning_rate": 2.1727941176470588e-05,
"loss": 0.7909,
"step": 1500
},
{
"epoch": 3.68,
"learning_rate": 1.8970588235294116e-05,
"loss": 0.6647,
"step": 2000
},
{
"epoch": 4.6,
"learning_rate": 1.6213235294117647e-05,
"loss": 0.5609,
"step": 2500
},
{
"epoch": 5.51,
"learning_rate": 1.3455882352941177e-05,
"loss": 0.4804,
"step": 3000
},
{
"epoch": 6.43,
"learning_rate": 1.0698529411764707e-05,
"loss": 0.4185,
"step": 3500
},
{
"epoch": 7.35,
"learning_rate": 7.941176470588236e-06,
"loss": 0.3687,
"step": 4000
},
{
"epoch": 8.27,
"learning_rate": 5.183823529411765e-06,
"loss": 0.3305,
"step": 4500
},
{
"epoch": 9.19,
"learning_rate": 2.4264705882352943e-06,
"loss": 0.3006,
"step": 5000
},
{
"epoch": 10.0,
"step": 5440,
"total_flos": 3.411490418678825e+17,
"train_loss": 0.6234890404869529,
"train_runtime": 1983.4548,
"train_samples_per_second": 658.165,
"train_steps_per_second": 2.743
}
],
"max_steps": 5440,
"num_train_epochs": 10,
"total_flos": 3.411490418678825e+17,
"trial_name": null,
"trial_params": null
}