xuancoblab2023's picture
Training in progress, epoch 2
48bc629 verified
raw
history blame
1.69 kB
{
"best_metric": 0.824853228962818,
"best_model_checkpoint": "tiny-bert-sst2-distilled/run-10/checkpoint-192",
"epoch": 2.0,
"eval_steps": 500,
"global_step": 192,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 1.9170680046081543,
"learning_rate": 0.0005212663608648181,
"loss": 0.5641,
"step": 96
},
{
"epoch": 1.0,
"eval_accuracy": 0.799412915851272,
"eval_f1": 0.8197009674582234,
"eval_loss": 0.47259676456451416,
"eval_precision": 0.744408945686901,
"eval_recall": 0.9119373776908023,
"eval_runtime": 30.8424,
"eval_samples_per_second": 33.136,
"eval_steps_per_second": 1.038,
"step": 96
},
{
"epoch": 2.0,
"grad_norm": 2.4848690032958984,
"learning_rate": 0.00026063318043240905,
"loss": 0.456,
"step": 192
},
{
"epoch": 2.0,
"eval_accuracy": 0.824853228962818,
"eval_f1": 0.8406055209260909,
"eval_loss": 0.4349122643470764,
"eval_precision": 0.7712418300653595,
"eval_recall": 0.923679060665362,
"eval_runtime": 31.1515,
"eval_samples_per_second": 32.807,
"eval_steps_per_second": 1.027,
"step": 192
}
],
"logging_steps": 500,
"max_steps": 288,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"total_flos": 471390394560.0,
"train_batch_size": 32,
"trial_name": null,
"trial_params": {
"alpha": 0.9610512963031814,
"learning_rate": 0.0007818995412972273,
"num_train_epochs": 3,
"temperature": 24
}
}