devrim's picture
Pushing AE model.
4350817
{
"best_metric": 0.25107308099671105,
"best_model_checkpoint": "/opt/ml/checkpoints/checkpoint-14592",
"epoch": 2.666179426274438,
"global_step": 14592,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 3.045685279187818e-08,
"loss": 2.0931,
"step": 1
},
{
"epoch": 0.33,
"learning_rate": 2.8369184916418297e-05,
"loss": 0.2097,
"step": 1824
},
{
"epoch": 0.33,
"eval_ANS_f1": 0.1965132827324478,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.28789019516997744,
"eval_ANS_recall": 0.14916729182295574,
"eval_loss": 0.1584259271621704,
"eval_overall_accuracy": 0.9525067851669007,
"eval_overall_f1": 0.1965132827324478,
"eval_overall_precision": 0.28789019516997744,
"eval_overall_recall": 0.14916729182295574,
"eval_runtime": 271.0381,
"eval_samples_per_second": 64.637,
"step": 1824
},
{
"epoch": 0.67,
"learning_rate": 2.4823765712064277e-05,
"loss": 0.1427,
"step": 3648
},
{
"epoch": 0.67,
"eval_ANS_f1": 0.14480276621268787,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.3350021767522856,
"eval_ANS_recall": 0.09236309077269317,
"eval_loss": 0.1570661962032318,
"eval_overall_accuracy": 0.9537896342872352,
"eval_overall_f1": 0.14480276621268787,
"eval_overall_precision": 0.3350021767522856,
"eval_overall_recall": 0.09236309077269317,
"eval_runtime": 272.0666,
"eval_samples_per_second": 64.392,
"step": 3648
},
{
"epoch": 1.0,
"learning_rate": 2.127834650771025e-05,
"loss": 0.1381,
"step": 5472
},
{
"epoch": 1.0,
"eval_ANS_f1": 0.18933628222641427,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.34125371267781773,
"eval_ANS_recall": 0.13101275318829708,
"eval_loss": 0.15228605270385742,
"eval_overall_accuracy": 0.954028691761447,
"eval_overall_f1": 0.18933628222641427,
"eval_overall_precision": 0.34125371267781773,
"eval_overall_recall": 0.13101275318829708,
"eval_runtime": 272.4342,
"eval_samples_per_second": 64.305,
"step": 5472
},
{
"epoch": 1.33,
"learning_rate": 1.773292730335623e-05,
"loss": 0.1295,
"step": 7296
},
{
"epoch": 1.33,
"eval_ANS_f1": 0.18266756939742723,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.36841147018661813,
"eval_ANS_recall": 0.1214403600900225,
"eval_loss": 0.15274310111999512,
"eval_overall_accuracy": 0.9544393980616191,
"eval_overall_f1": 0.18266756939742723,
"eval_overall_precision": 0.36841147018661813,
"eval_overall_recall": 0.1214403600900225,
"eval_runtime": 269.6146,
"eval_samples_per_second": 64.978,
"step": 7296
},
{
"epoch": 1.67,
"learning_rate": 1.4187508099002203e-05,
"loss": 0.1275,
"step": 9120
},
{
"epoch": 1.67,
"eval_ANS_f1": 0.22205431725643288,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.3693876125968857,
"eval_ANS_recall": 0.1587396849212303,
"eval_loss": 0.15151338279247284,
"eval_overall_accuracy": 0.9544824840017387,
"eval_overall_f1": 0.22205431725643288,
"eval_overall_precision": 0.3693876125968857,
"eval_overall_recall": 0.1587396849212303,
"eval_runtime": 273.5764,
"eval_samples_per_second": 64.037,
"step": 9120
},
{
"epoch": 2.0,
"learning_rate": 1.0642088894648179e-05,
"loss": 0.1262,
"step": 10944
},
{
"epoch": 2.0,
"eval_ANS_f1": 0.22321074285236933,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.3670833905756285,
"eval_ANS_recall": 0.16036009002250562,
"eval_loss": 0.14961716532707214,
"eval_overall_accuracy": 0.9540054114550921,
"eval_overall_f1": 0.22321074285236933,
"eval_overall_precision": 0.3670833905756285,
"eval_overall_recall": 0.16036009002250562,
"eval_runtime": 271.8734,
"eval_samples_per_second": 64.438,
"step": 10944
},
{
"epoch": 2.33,
"learning_rate": 7.096669690294156e-06,
"loss": 0.1126,
"step": 12768
},
{
"epoch": 2.33,
"eval_ANS_f1": 0.2386963952765693,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.33832516654737654,
"eval_ANS_recall": 0.18439609902475618,
"eval_loss": 0.15988405048847198,
"eval_overall_accuracy": 0.9519723805225143,
"eval_overall_f1": 0.2386963952765693,
"eval_overall_precision": 0.33832516654737654,
"eval_overall_recall": 0.18439609902475618,
"eval_runtime": 273.7781,
"eval_samples_per_second": 63.99,
"step": 12768
},
{
"epoch": 2.67,
"learning_rate": 3.551250485940132e-06,
"loss": 0.1109,
"step": 14592
},
{
"epoch": 2.67,
"eval_ANS_f1": 0.25107308099671105,
"eval_ANS_number": 33325,
"eval_ANS_precision": 0.3296896349795042,
"eval_ANS_recall": 0.20273068267066766,
"eval_loss": 0.16195876896381378,
"eval_overall_accuracy": 0.9512107322907231,
"eval_overall_f1": 0.25107308099671105,
"eval_overall_precision": 0.3296896349795042,
"eval_overall_recall": 0.20273068267066766,
"eval_runtime": 275.1376,
"eval_samples_per_second": 63.674,
"step": 14592
}
],
"max_steps": 16419,
"num_train_epochs": 3,
"total_flos": 3.702162559715073e+17,
"trial_name": null,
"trial_params": null
}