atomic-roberta-large-full / trainer_state.json
phosseini's picture
Upload trainer_state.json
fb83b10
{
"best_metric": 0.8793088793754578,
"best_model_checkpoint": "models/checkpoints/checkpoint-5500",
"epoch": 0.4509582863585118,
"global_step": 5500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01,
"learning_rate": 1.998360118071499e-05,
"loss": 1.518,
"step": 100
},
{
"epoch": 0.02,
"learning_rate": 1.9967202361429978e-05,
"loss": 1.2325,
"step": 200
},
{
"epoch": 0.02,
"learning_rate": 1.9950803542144966e-05,
"loss": 1.1664,
"step": 300
},
{
"epoch": 0.03,
"learning_rate": 1.9934404722859955e-05,
"loss": 1.1746,
"step": 400
},
{
"epoch": 0.04,
"learning_rate": 1.9918005903574943e-05,
"loss": 1.1251,
"step": 500
},
{
"epoch": 0.04,
"eval_loss": 0.964438259601593,
"eval_runtime": 267.7236,
"eval_samples_per_second": 255.906,
"eval_steps_per_second": 15.994,
"step": 500
},
{
"epoch": 0.05,
"learning_rate": 1.9901607084289934e-05,
"loss": 1.1041,
"step": 600
},
{
"epoch": 0.06,
"learning_rate": 1.988520826500492e-05,
"loss": 1.0817,
"step": 700
},
{
"epoch": 0.07,
"learning_rate": 1.986880944571991e-05,
"loss": 1.0594,
"step": 800
},
{
"epoch": 0.07,
"learning_rate": 1.98524106264349e-05,
"loss": 1.0642,
"step": 900
},
{
"epoch": 0.08,
"learning_rate": 1.9836011807149887e-05,
"loss": 1.0453,
"step": 1000
},
{
"epoch": 0.08,
"eval_loss": 0.9285051226615906,
"eval_runtime": 253.7626,
"eval_samples_per_second": 269.985,
"eval_steps_per_second": 16.874,
"step": 1000
},
{
"epoch": 0.09,
"learning_rate": 1.9819612987864875e-05,
"loss": 1.0486,
"step": 1100
},
{
"epoch": 0.1,
"learning_rate": 1.9803214168579864e-05,
"loss": 1.0549,
"step": 1200
},
{
"epoch": 0.11,
"learning_rate": 1.9786815349294852e-05,
"loss": 1.0639,
"step": 1300
},
{
"epoch": 0.11,
"learning_rate": 1.977041653000984e-05,
"loss": 1.0354,
"step": 1400
},
{
"epoch": 0.12,
"learning_rate": 1.9754017710724828e-05,
"loss": 1.0393,
"step": 1500
},
{
"epoch": 0.12,
"eval_loss": 0.9202534556388855,
"eval_runtime": 264.8239,
"eval_samples_per_second": 258.708,
"eval_steps_per_second": 16.169,
"step": 1500
},
{
"epoch": 0.13,
"learning_rate": 1.9737618891439817e-05,
"loss": 1.0381,
"step": 1600
},
{
"epoch": 0.14,
"learning_rate": 1.9721220072154808e-05,
"loss": 1.0672,
"step": 1700
},
{
"epoch": 0.15,
"learning_rate": 1.9704821252869793e-05,
"loss": 1.0335,
"step": 1800
},
{
"epoch": 0.16,
"learning_rate": 1.9688422433584785e-05,
"loss": 1.0239,
"step": 1900
},
{
"epoch": 0.16,
"learning_rate": 1.9672023614299773e-05,
"loss": 1.0012,
"step": 2000
},
{
"epoch": 0.16,
"eval_loss": 0.9098692536354065,
"eval_runtime": 264.5198,
"eval_samples_per_second": 259.005,
"eval_steps_per_second": 16.188,
"step": 2000
},
{
"epoch": 0.17,
"learning_rate": 1.965562479501476e-05,
"loss": 1.0146,
"step": 2100
},
{
"epoch": 0.18,
"learning_rate": 1.963922597572975e-05,
"loss": 1.0502,
"step": 2200
},
{
"epoch": 0.19,
"learning_rate": 1.9622827156444737e-05,
"loss": 1.0062,
"step": 2300
},
{
"epoch": 0.2,
"learning_rate": 1.9606428337159726e-05,
"loss": 1.0144,
"step": 2400
},
{
"epoch": 0.2,
"learning_rate": 1.9590029517874714e-05,
"loss": 0.9826,
"step": 2500
},
{
"epoch": 0.2,
"eval_loss": 0.9166184067726135,
"eval_runtime": 256.5893,
"eval_samples_per_second": 267.01,
"eval_steps_per_second": 16.688,
"step": 2500
},
{
"epoch": 0.21,
"learning_rate": 1.9573630698589702e-05,
"loss": 1.0269,
"step": 2600
},
{
"epoch": 0.22,
"learning_rate": 1.9557231879304694e-05,
"loss": 0.9898,
"step": 2700
},
{
"epoch": 0.23,
"learning_rate": 1.954083306001968e-05,
"loss": 1.0248,
"step": 2800
},
{
"epoch": 0.24,
"learning_rate": 1.952443424073467e-05,
"loss": 0.9996,
"step": 2900
},
{
"epoch": 0.25,
"learning_rate": 1.950803542144966e-05,
"loss": 0.9595,
"step": 3000
},
{
"epoch": 0.25,
"eval_loss": 0.9191610217094421,
"eval_runtime": 272.1388,
"eval_samples_per_second": 251.754,
"eval_steps_per_second": 15.735,
"step": 3000
},
{
"epoch": 0.25,
"learning_rate": 1.9491636602164647e-05,
"loss": 0.9873,
"step": 3100
},
{
"epoch": 0.26,
"learning_rate": 1.9475237782879635e-05,
"loss": 0.9667,
"step": 3200
},
{
"epoch": 0.27,
"learning_rate": 1.9458838963594623e-05,
"loss": 0.9666,
"step": 3300
},
{
"epoch": 0.28,
"learning_rate": 1.944244014430961e-05,
"loss": 1.0102,
"step": 3400
},
{
"epoch": 0.29,
"learning_rate": 1.94260413250246e-05,
"loss": 0.9511,
"step": 3500
},
{
"epoch": 0.29,
"eval_loss": 0.9013209342956543,
"eval_runtime": 267.0529,
"eval_samples_per_second": 256.548,
"eval_steps_per_second": 16.034,
"step": 3500
},
{
"epoch": 0.3,
"learning_rate": 1.9409642505739588e-05,
"loss": 0.9927,
"step": 3600
},
{
"epoch": 0.3,
"learning_rate": 1.9393243686454576e-05,
"loss": 0.9507,
"step": 3700
},
{
"epoch": 0.31,
"learning_rate": 1.9376844867169567e-05,
"loss": 1.0049,
"step": 3800
},
{
"epoch": 0.32,
"learning_rate": 1.9360446047884552e-05,
"loss": 0.9385,
"step": 3900
},
{
"epoch": 0.33,
"learning_rate": 1.9344047228599544e-05,
"loss": 0.9644,
"step": 4000
},
{
"epoch": 0.33,
"eval_loss": 0.8957135081291199,
"eval_runtime": 260.7988,
"eval_samples_per_second": 262.701,
"eval_steps_per_second": 16.419,
"step": 4000
},
{
"epoch": 0.34,
"learning_rate": 1.932764840931453e-05,
"loss": 0.9293,
"step": 4100
},
{
"epoch": 0.34,
"learning_rate": 1.931124959002952e-05,
"loss": 0.9298,
"step": 4200
},
{
"epoch": 0.35,
"learning_rate": 1.929485077074451e-05,
"loss": 0.9218,
"step": 4300
},
{
"epoch": 0.36,
"learning_rate": 1.9278451951459497e-05,
"loss": 0.9523,
"step": 4400
},
{
"epoch": 0.37,
"learning_rate": 1.9262053132174485e-05,
"loss": 0.9079,
"step": 4500
},
{
"epoch": 0.37,
"eval_loss": 0.9050074815750122,
"eval_runtime": 264.4439,
"eval_samples_per_second": 259.079,
"eval_steps_per_second": 16.192,
"step": 4500
},
{
"epoch": 0.38,
"learning_rate": 1.9245654312889473e-05,
"loss": 0.9437,
"step": 4600
},
{
"epoch": 0.39,
"learning_rate": 1.922925549360446e-05,
"loss": 0.9395,
"step": 4700
},
{
"epoch": 0.39,
"learning_rate": 1.921285667431945e-05,
"loss": 0.969,
"step": 4800
},
{
"epoch": 0.4,
"learning_rate": 1.9196457855034438e-05,
"loss": 0.926,
"step": 4900
},
{
"epoch": 0.41,
"learning_rate": 1.9180059035749426e-05,
"loss": 0.9178,
"step": 5000
},
{
"epoch": 0.41,
"eval_loss": 0.8956226706504822,
"eval_runtime": 259.9134,
"eval_samples_per_second": 263.595,
"eval_steps_per_second": 16.475,
"step": 5000
},
{
"epoch": 0.42,
"learning_rate": 1.9163660216464418e-05,
"loss": 0.9697,
"step": 5100
},
{
"epoch": 0.43,
"learning_rate": 1.9147261397179403e-05,
"loss": 0.9448,
"step": 5200
},
{
"epoch": 0.43,
"learning_rate": 1.9130862577894394e-05,
"loss": 0.9118,
"step": 5300
},
{
"epoch": 0.44,
"learning_rate": 1.911446375860938e-05,
"loss": 0.9321,
"step": 5400
},
{
"epoch": 0.45,
"learning_rate": 1.909806493932437e-05,
"loss": 0.9289,
"step": 5500
},
{
"epoch": 0.45,
"eval_loss": 0.8793088793754578,
"eval_runtime": 261.4397,
"eval_samples_per_second": 262.057,
"eval_steps_per_second": 16.379,
"step": 5500
}
],
"max_steps": 121960,
"num_train_epochs": 10,
"total_flos": 1.922426903232e+16,
"trial_name": null,
"trial_params": null
}