bert-base-uncased-squad-v1 / trainer_state.json
helenai's picture
Add PyTorch model
bd3d859
raw
history blame
4.13 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"global_step": 14754,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.07,
"learning_rate": 2.8983326555510372e-05,
"loss": 2.4547,
"step": 500
},
{
"epoch": 0.14,
"learning_rate": 2.796665311102074e-05,
"loss": 1.4963,
"step": 1000
},
{
"epoch": 0.2,
"learning_rate": 2.694997966653111e-05,
"loss": 1.4012,
"step": 1500
},
{
"epoch": 0.27,
"learning_rate": 2.593330622204148e-05,
"loss": 1.3187,
"step": 2000
},
{
"epoch": 0.34,
"learning_rate": 2.491663277755185e-05,
"loss": 1.249,
"step": 2500
},
{
"epoch": 0.41,
"learning_rate": 2.389995933306222e-05,
"loss": 1.1969,
"step": 3000
},
{
"epoch": 0.47,
"learning_rate": 2.288328588857259e-05,
"loss": 1.1941,
"step": 3500
},
{
"epoch": 0.54,
"learning_rate": 2.1866612444082963e-05,
"loss": 1.1202,
"step": 4000
},
{
"epoch": 0.61,
"learning_rate": 2.084993899959333e-05,
"loss": 1.0896,
"step": 4500
},
{
"epoch": 0.68,
"learning_rate": 1.9833265555103702e-05,
"loss": 1.0834,
"step": 5000
},
{
"epoch": 0.75,
"learning_rate": 1.8816592110614073e-05,
"loss": 1.1031,
"step": 5500
},
{
"epoch": 0.81,
"learning_rate": 1.779991866612444e-05,
"loss": 1.0656,
"step": 6000
},
{
"epoch": 0.88,
"learning_rate": 1.6783245221634812e-05,
"loss": 1.0266,
"step": 6500
},
{
"epoch": 0.95,
"learning_rate": 1.576657177714518e-05,
"loss": 1.0489,
"step": 7000
},
{
"epoch": 1.02,
"learning_rate": 1.4749898332655551e-05,
"loss": 0.944,
"step": 7500
},
{
"epoch": 1.08,
"learning_rate": 1.373322488816592e-05,
"loss": 0.7275,
"step": 8000
},
{
"epoch": 1.15,
"learning_rate": 1.271655144367629e-05,
"loss": 0.7439,
"step": 8500
},
{
"epoch": 1.22,
"learning_rate": 1.1699877999186661e-05,
"loss": 0.7352,
"step": 9000
},
{
"epoch": 1.29,
"learning_rate": 1.0683204554697033e-05,
"loss": 0.693,
"step": 9500
},
{
"epoch": 1.36,
"learning_rate": 9.666531110207402e-06,
"loss": 0.736,
"step": 10000
},
{
"epoch": 1.42,
"learning_rate": 8.649857665717772e-06,
"loss": 0.7238,
"step": 10500
},
{
"epoch": 1.49,
"learning_rate": 7.633184221228141e-06,
"loss": 0.7266,
"step": 11000
},
{
"epoch": 1.56,
"learning_rate": 6.616510776738511e-06,
"loss": 0.7112,
"step": 11500
},
{
"epoch": 1.63,
"learning_rate": 5.5998373322488825e-06,
"loss": 0.7348,
"step": 12000
},
{
"epoch": 1.69,
"learning_rate": 4.583163887759252e-06,
"loss": 0.6808,
"step": 12500
},
{
"epoch": 1.76,
"learning_rate": 3.566490443269622e-06,
"loss": 0.6928,
"step": 13000
},
{
"epoch": 1.83,
"learning_rate": 2.549816998779992e-06,
"loss": 0.7016,
"step": 13500
},
{
"epoch": 1.9,
"learning_rate": 1.533143554290362e-06,
"loss": 0.719,
"step": 14000
},
{
"epoch": 1.97,
"learning_rate": 5.164701098007319e-07,
"loss": 0.7163,
"step": 14500
},
{
"epoch": 2.0,
"step": 14754,
"total_flos": 3.4696551139946496e+16,
"train_loss": 0.9892611550787075,
"train_runtime": 3658.4093,
"train_samples_per_second": 48.395,
"train_steps_per_second": 4.033
}
],
"max_steps": 14754,
"num_train_epochs": 2,
"total_flos": 3.4696551139946496e+16,
"trial_name": null,
"trial_params": null
}