colerobertson's picture
Training in progress, epoch 1
d89240d verified
raw
history blame
4.09 kB
{
"best_metric": 0.7326732673267327,
"best_model_checkpoint": "distilhubert-finetuned-not-a-word2/run-0/checkpoint-48",
"epoch": 2.0,
"eval_steps": 500,
"global_step": 96,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.1,
"grad_norm": 1.2701194286346436,
"learning_rate": 1.7573936619349767e-06,
"loss": 0.7024,
"step": 5
},
{
"epoch": 0.21,
"grad_norm": 1.218135952949524,
"learning_rate": 3.5147873238699533e-06,
"loss": 0.6932,
"step": 10
},
{
"epoch": 0.31,
"grad_norm": 1.5910921096801758,
"learning_rate": 5.2721809858049295e-06,
"loss": 0.6916,
"step": 15
},
{
"epoch": 0.42,
"grad_norm": 1.0592715740203857,
"learning_rate": 7.029574647739907e-06,
"loss": 0.6854,
"step": 20
},
{
"epoch": 0.52,
"grad_norm": 1.132177472114563,
"learning_rate": 8.786968309674883e-06,
"loss": 0.6719,
"step": 25
},
{
"epoch": 0.62,
"grad_norm": 0.9851714372634888,
"learning_rate": 1.0544361971609859e-05,
"loss": 0.6633,
"step": 30
},
{
"epoch": 0.73,
"grad_norm": 1.5592440366744995,
"learning_rate": 1.2301755633544835e-05,
"loss": 0.6288,
"step": 35
},
{
"epoch": 0.83,
"grad_norm": 1.8793619871139526,
"learning_rate": 1.4059149295479813e-05,
"loss": 0.6561,
"step": 40
},
{
"epoch": 0.94,
"grad_norm": 1.987308382987976,
"learning_rate": 1.581654295741479e-05,
"loss": 0.6343,
"step": 45
},
{
"epoch": 1.0,
"eval_f1": 0.7326732673267327,
"eval_loss": 0.6868629455566406,
"eval_runtime": 1.3443,
"eval_samples_per_second": 47.609,
"eval_steps_per_second": 5.951,
"step": 48
},
{
"epoch": 1.04,
"grad_norm": 3.1652989387512207,
"learning_rate": 1.6792872769600888e-05,
"loss": 0.6712,
"step": 50
},
{
"epoch": 1.15,
"grad_norm": 0.9990371465682983,
"learning_rate": 1.6597606807163667e-05,
"loss": 0.6657,
"step": 55
},
{
"epoch": 1.25,
"grad_norm": 1.8210965394973755,
"learning_rate": 1.640234084472645e-05,
"loss": 0.476,
"step": 60
},
{
"epoch": 1.35,
"grad_norm": 0.5364782810211182,
"learning_rate": 1.6207074882289228e-05,
"loss": 0.5768,
"step": 65
},
{
"epoch": 1.46,
"grad_norm": 1.6772537231445312,
"learning_rate": 1.6011808919852007e-05,
"loss": 0.5039,
"step": 70
},
{
"epoch": 1.56,
"grad_norm": 1.0453933477401733,
"learning_rate": 1.581654295741479e-05,
"loss": 0.7182,
"step": 75
},
{
"epoch": 1.67,
"grad_norm": 1.621067762374878,
"learning_rate": 1.562127699497757e-05,
"loss": 0.6178,
"step": 80
},
{
"epoch": 1.77,
"grad_norm": 1.7878597974777222,
"learning_rate": 1.542601103254035e-05,
"loss": 0.6192,
"step": 85
},
{
"epoch": 1.88,
"grad_norm": 1.6824157238006592,
"learning_rate": 1.5230745070103131e-05,
"loss": 0.6109,
"step": 90
},
{
"epoch": 1.98,
"grad_norm": 0.7989784479141235,
"learning_rate": 1.5035479107665912e-05,
"loss": 0.6367,
"step": 95
},
{
"epoch": 2.0,
"eval_f1": 0.7326732673267327,
"eval_loss": 0.7258987426757812,
"eval_runtime": 1.3947,
"eval_samples_per_second": 45.888,
"eval_steps_per_second": 5.736,
"step": 96
}
],
"logging_steps": 5,
"max_steps": 480,
"num_input_tokens_seen": 0,
"num_train_epochs": 10,
"save_steps": 500,
"total_flos": 1442567462539200.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": {
"learning_rate": 1.6870979154575775e-05,
"per_device_train_batch_size": 4
}
}