designTrends / checkpoint-36 /trainer_state.json
coan's picture
Upload folder using huggingface_hub
3252aa1 verified
{
"best_metric": 0.2903565764427185,
"best_model_checkpoint": "designTrends/checkpoint-36",
"epoch": 2.0,
"eval_steps": 500,
"global_step": 36,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05555555555555555,
"grad_norm": 31.24921989440918,
"learning_rate": 8.333333333333334e-06,
"loss": 1.6824,
"step": 1
},
{
"epoch": 0.1111111111111111,
"grad_norm": 20.90537452697754,
"learning_rate": 1.6666666666666667e-05,
"loss": 1.1182,
"step": 2
},
{
"epoch": 0.16666666666666666,
"grad_norm": 37.51194763183594,
"learning_rate": 2.5e-05,
"loss": 2.1944,
"step": 3
},
{
"epoch": 0.2222222222222222,
"grad_norm": 30.805044174194336,
"learning_rate": 3.3333333333333335e-05,
"loss": 1.8025,
"step": 4
},
{
"epoch": 0.2777777777777778,
"grad_norm": 15.017728805541992,
"learning_rate": 4.166666666666667e-05,
"loss": 0.4371,
"step": 5
},
{
"epoch": 0.3333333333333333,
"grad_norm": 37.36845016479492,
"learning_rate": 5e-05,
"loss": 2.3815,
"step": 6
},
{
"epoch": 0.3888888888888889,
"grad_norm": 36.02885818481445,
"learning_rate": 4.8958333333333335e-05,
"loss": 1.3463,
"step": 7
},
{
"epoch": 0.4444444444444444,
"grad_norm": 34.15700149536133,
"learning_rate": 4.791666666666667e-05,
"loss": 2.0606,
"step": 8
},
{
"epoch": 0.5,
"grad_norm": 30.92742347717285,
"learning_rate": 4.6875e-05,
"loss": 1.2667,
"step": 9
},
{
"epoch": 0.5555555555555556,
"grad_norm": 17.669174194335938,
"learning_rate": 4.5833333333333334e-05,
"loss": 0.7387,
"step": 10
},
{
"epoch": 0.6111111111111112,
"grad_norm": 21.606388092041016,
"learning_rate": 4.4791666666666673e-05,
"loss": 1.0753,
"step": 11
},
{
"epoch": 0.6666666666666666,
"grad_norm": 11.96320629119873,
"learning_rate": 4.375e-05,
"loss": 0.4321,
"step": 12
},
{
"epoch": 0.7222222222222222,
"grad_norm": 28.005523681640625,
"learning_rate": 4.270833333333333e-05,
"loss": 1.7522,
"step": 13
},
{
"epoch": 0.7777777777777778,
"grad_norm": 13.161625862121582,
"learning_rate": 4.166666666666667e-05,
"loss": 0.4633,
"step": 14
},
{
"epoch": 0.8333333333333334,
"grad_norm": 30.44942283630371,
"learning_rate": 4.0625000000000005e-05,
"loss": 1.2798,
"step": 15
},
{
"epoch": 0.8888888888888888,
"grad_norm": 18.832170486450195,
"learning_rate": 3.958333333333333e-05,
"loss": 0.6956,
"step": 16
},
{
"epoch": 0.9444444444444444,
"grad_norm": 8.636096000671387,
"learning_rate": 3.854166666666667e-05,
"loss": 0.2504,
"step": 17
},
{
"epoch": 1.0,
"grad_norm": 4.477694511413574,
"learning_rate": 3.7500000000000003e-05,
"loss": 0.0766,
"step": 18
},
{
"epoch": 1.0,
"eval_accuracy": 0.7777777777777778,
"eval_f1_macro": 0.6875,
"eval_f1_micro": 0.7777777777777778,
"eval_f1_weighted": 0.6944444444444444,
"eval_loss": 1.071279525756836,
"eval_precision_macro": 0.65,
"eval_precision_micro": 0.7777777777777778,
"eval_precision_weighted": 0.6444444444444444,
"eval_recall_macro": 0.75,
"eval_recall_micro": 0.7777777777777778,
"eval_recall_weighted": 0.7777777777777778,
"eval_runtime": 10.4277,
"eval_samples_per_second": 0.863,
"eval_steps_per_second": 0.288,
"step": 18
},
{
"epoch": 1.0555555555555556,
"grad_norm": 11.063207626342773,
"learning_rate": 3.6458333333333336e-05,
"loss": 0.2519,
"step": 19
},
{
"epoch": 1.1111111111111112,
"grad_norm": 11.03740406036377,
"learning_rate": 3.541666666666667e-05,
"loss": 0.2958,
"step": 20
},
{
"epoch": 1.1666666666666667,
"grad_norm": 1.8288331031799316,
"learning_rate": 3.4375e-05,
"loss": 0.0406,
"step": 21
},
{
"epoch": 1.2222222222222223,
"grad_norm": 31.19927215576172,
"learning_rate": 3.3333333333333335e-05,
"loss": 1.0578,
"step": 22
},
{
"epoch": 1.2777777777777777,
"grad_norm": 5.91452693939209,
"learning_rate": 3.229166666666667e-05,
"loss": 0.1396,
"step": 23
},
{
"epoch": 1.3333333333333333,
"grad_norm": 28.180068969726562,
"learning_rate": 3.125e-05,
"loss": 0.9014,
"step": 24
},
{
"epoch": 1.3888888888888888,
"grad_norm": 6.710000038146973,
"learning_rate": 3.0208333333333334e-05,
"loss": 0.1434,
"step": 25
},
{
"epoch": 1.4444444444444444,
"grad_norm": 1.4782997369766235,
"learning_rate": 2.916666666666667e-05,
"loss": 0.033,
"step": 26
},
{
"epoch": 1.5,
"grad_norm": 10.041507720947266,
"learning_rate": 2.8125000000000003e-05,
"loss": 0.2318,
"step": 27
},
{
"epoch": 1.5555555555555556,
"grad_norm": 15.513708114624023,
"learning_rate": 2.7083333333333332e-05,
"loss": 0.4245,
"step": 28
},
{
"epoch": 1.6111111111111112,
"grad_norm": 5.358726978302002,
"learning_rate": 2.604166666666667e-05,
"loss": 0.1601,
"step": 29
},
{
"epoch": 1.6666666666666665,
"grad_norm": 1.7197539806365967,
"learning_rate": 2.5e-05,
"loss": 0.042,
"step": 30
},
{
"epoch": 1.7222222222222223,
"grad_norm": 0.5361698865890503,
"learning_rate": 2.3958333333333334e-05,
"loss": 0.0149,
"step": 31
},
{
"epoch": 1.7777777777777777,
"grad_norm": 4.341958522796631,
"learning_rate": 2.2916666666666667e-05,
"loss": 0.1039,
"step": 32
},
{
"epoch": 1.8333333333333335,
"grad_norm": 20.135475158691406,
"learning_rate": 2.1875e-05,
"loss": 0.5532,
"step": 33
},
{
"epoch": 1.8888888888888888,
"grad_norm": 0.5918666124343872,
"learning_rate": 2.0833333333333336e-05,
"loss": 0.0155,
"step": 34
},
{
"epoch": 1.9444444444444444,
"grad_norm": 2.947694778442383,
"learning_rate": 1.9791666666666665e-05,
"loss": 0.0862,
"step": 35
},
{
"epoch": 2.0,
"grad_norm": 0.9083338379859924,
"learning_rate": 1.8750000000000002e-05,
"loss": 0.0167,
"step": 36
},
{
"epoch": 2.0,
"eval_accuracy": 0.8888888888888888,
"eval_f1_macro": 0.8999999999999999,
"eval_f1_micro": 0.8888888888888888,
"eval_f1_weighted": 0.8888888888888888,
"eval_loss": 0.2903565764427185,
"eval_precision_macro": 0.9166666666666666,
"eval_precision_micro": 0.8888888888888888,
"eval_precision_weighted": 0.9259259259259258,
"eval_recall_macro": 0.9166666666666666,
"eval_recall_micro": 0.8888888888888888,
"eval_recall_weighted": 0.8888888888888888,
"eval_runtime": 9.5629,
"eval_samples_per_second": 0.941,
"eval_steps_per_second": 0.314,
"step": 36
}
],
"logging_steps": 1,
"max_steps": 54,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 5,
"early_stopping_threshold": 0.01
},
"attributes": {
"early_stopping_patience_counter": 0
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 1.917552219512832e+16,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}