|
{ |
|
"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 |
|
} |
|
} |
|
|