clip-mixer-300k / trainer_state.json
bilbo991's picture
End of training
5fdcb7d
Raw
History Blame Contribute Delete
8.04 kB
{
"best_metric": 2.4178671836853027,
"best_model_checkpoint": "clip-mixer-300k/checkpoint-18750",
"epoch": 3.0,
"global_step": 28125,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05,
"learning_rate": 4.9111111111111114e-05,
"loss": 2.1259,
"step": 500
},
{
"epoch": 0.11,
"learning_rate": 4.8222222222222225e-05,
"loss": 1.8445,
"step": 1000
},
{
"epoch": 0.16,
"learning_rate": 4.7333333333333336e-05,
"loss": 1.7594,
"step": 1500
},
{
"epoch": 0.21,
"learning_rate": 4.644444444444445e-05,
"loss": 1.7145,
"step": 2000
},
{
"epoch": 0.27,
"learning_rate": 4.555555555555556e-05,
"loss": 1.6367,
"step": 2500
},
{
"epoch": 0.32,
"learning_rate": 4.466666666666667e-05,
"loss": 1.6162,
"step": 3000
},
{
"epoch": 0.37,
"learning_rate": 4.377777777777778e-05,
"loss": 1.5544,
"step": 3500
},
{
"epoch": 0.43,
"learning_rate": 4.2888888888888886e-05,
"loss": 1.5502,
"step": 4000
},
{
"epoch": 0.48,
"learning_rate": 4.2e-05,
"loss": 1.5128,
"step": 4500
},
{
"epoch": 0.53,
"learning_rate": 4.111111111111111e-05,
"loss": 1.473,
"step": 5000
},
{
"epoch": 0.59,
"learning_rate": 4.022222222222222e-05,
"loss": 1.4328,
"step": 5500
},
{
"epoch": 0.64,
"learning_rate": 3.933333333333333e-05,
"loss": 1.4208,
"step": 6000
},
{
"epoch": 0.69,
"learning_rate": 3.844444444444444e-05,
"loss": 1.4009,
"step": 6500
},
{
"epoch": 0.75,
"learning_rate": 3.7555555555555554e-05,
"loss": 1.3752,
"step": 7000
},
{
"epoch": 0.8,
"learning_rate": 3.6666666666666666e-05,
"loss": 1.3387,
"step": 7500
},
{
"epoch": 0.85,
"learning_rate": 3.577777777777778e-05,
"loss": 1.3141,
"step": 8000
},
{
"epoch": 0.91,
"learning_rate": 3.4888888888888895e-05,
"loss": 1.3023,
"step": 8500
},
{
"epoch": 0.96,
"learning_rate": 3.4000000000000007e-05,
"loss": 1.2811,
"step": 9000
},
{
"epoch": 1.0,
"eval_loss": 2.599307060241699,
"eval_runtime": 359.2177,
"eval_samples_per_second": 83.515,
"eval_steps_per_second": 2.611,
"step": 9375
},
{
"epoch": 1.01,
"learning_rate": 3.311111111111112e-05,
"loss": 1.1965,
"step": 9500
},
{
"epoch": 1.07,
"learning_rate": 3.222222222222223e-05,
"loss": 0.9531,
"step": 10000
},
{
"epoch": 1.12,
"learning_rate": 3.1333333333333334e-05,
"loss": 0.9502,
"step": 10500
},
{
"epoch": 1.17,
"learning_rate": 3.044444444444445e-05,
"loss": 0.9351,
"step": 11000
},
{
"epoch": 1.23,
"learning_rate": 2.955555555555556e-05,
"loss": 0.9377,
"step": 11500
},
{
"epoch": 1.28,
"learning_rate": 2.8666666666666668e-05,
"loss": 0.9437,
"step": 12000
},
{
"epoch": 1.33,
"learning_rate": 2.777777777777778e-05,
"loss": 0.9403,
"step": 12500
},
{
"epoch": 1.39,
"learning_rate": 2.688888888888889e-05,
"loss": 0.9138,
"step": 13000
},
{
"epoch": 1.44,
"learning_rate": 2.6000000000000002e-05,
"loss": 0.9201,
"step": 13500
},
{
"epoch": 1.49,
"learning_rate": 2.5111111111111113e-05,
"loss": 0.8969,
"step": 14000
},
{
"epoch": 1.55,
"learning_rate": 2.4222222222222224e-05,
"loss": 0.9031,
"step": 14500
},
{
"epoch": 1.6,
"learning_rate": 2.3333333333333336e-05,
"loss": 0.8836,
"step": 15000
},
{
"epoch": 1.65,
"learning_rate": 2.2444444444444447e-05,
"loss": 0.8711,
"step": 15500
},
{
"epoch": 1.71,
"learning_rate": 2.1555555555555555e-05,
"loss": 0.8671,
"step": 16000
},
{
"epoch": 1.76,
"learning_rate": 2.0666666666666666e-05,
"loss": 0.8494,
"step": 16500
},
{
"epoch": 1.81,
"learning_rate": 1.9777777777777778e-05,
"loss": 0.8346,
"step": 17000
},
{
"epoch": 1.87,
"learning_rate": 1.888888888888889e-05,
"loss": 0.8396,
"step": 17500
},
{
"epoch": 1.92,
"learning_rate": 1.8e-05,
"loss": 0.802,
"step": 18000
},
{
"epoch": 1.97,
"learning_rate": 1.7111111111111112e-05,
"loss": 0.8159,
"step": 18500
},
{
"epoch": 2.0,
"eval_loss": 2.4178671836853027,
"eval_runtime": 360.9468,
"eval_samples_per_second": 83.115,
"eval_steps_per_second": 2.599,
"step": 18750
},
{
"epoch": 2.03,
"learning_rate": 1.6222222222222223e-05,
"loss": 0.637,
"step": 19000
},
{
"epoch": 2.08,
"learning_rate": 1.5333333333333334e-05,
"loss": 0.4617,
"step": 19500
},
{
"epoch": 2.13,
"learning_rate": 1.4444444444444444e-05,
"loss": 0.4634,
"step": 20000
},
{
"epoch": 2.19,
"learning_rate": 1.3555555555555557e-05,
"loss": 0.4644,
"step": 20500
},
{
"epoch": 2.24,
"learning_rate": 1.2666666666666668e-05,
"loss": 0.4426,
"step": 21000
},
{
"epoch": 2.29,
"learning_rate": 1.1777777777777778e-05,
"loss": 0.4559,
"step": 21500
},
{
"epoch": 2.35,
"learning_rate": 1.088888888888889e-05,
"loss": 0.443,
"step": 22000
},
{
"epoch": 2.4,
"learning_rate": 1e-05,
"loss": 0.4359,
"step": 22500
},
{
"epoch": 2.45,
"learning_rate": 9.111111111111112e-06,
"loss": 0.4303,
"step": 23000
},
{
"epoch": 2.51,
"learning_rate": 8.222222222222223e-06,
"loss": 0.4228,
"step": 23500
},
{
"epoch": 2.56,
"learning_rate": 7.333333333333334e-06,
"loss": 0.4125,
"step": 24000
},
{
"epoch": 2.61,
"learning_rate": 6.4444444444444445e-06,
"loss": 0.4139,
"step": 24500
},
{
"epoch": 2.67,
"learning_rate": 5.555555555555556e-06,
"loss": 0.4146,
"step": 25000
},
{
"epoch": 2.72,
"learning_rate": 4.666666666666667e-06,
"loss": 0.3986,
"step": 25500
},
{
"epoch": 2.77,
"learning_rate": 3.777777777777778e-06,
"loss": 0.4017,
"step": 26000
},
{
"epoch": 2.83,
"learning_rate": 2.888888888888889e-06,
"loss": 0.3922,
"step": 26500
},
{
"epoch": 2.88,
"learning_rate": 2.0000000000000003e-06,
"loss": 0.3917,
"step": 27000
},
{
"epoch": 2.93,
"learning_rate": 1.1111111111111112e-06,
"loss": 0.385,
"step": 27500
},
{
"epoch": 2.99,
"learning_rate": 2.2222222222222224e-07,
"loss": 0.374,
"step": 28000
},
{
"epoch": 3.0,
"eval_loss": 2.4597387313842773,
"eval_runtime": 361.6503,
"eval_samples_per_second": 82.953,
"eval_steps_per_second": 2.594,
"step": 28125
},
{
"epoch": 3.0,
"step": 28125,
"total_flos": 9.03625120512e+16,
"train_loss": 0.9465052639431424,
"train_runtime": 18848.6969,
"train_samples_per_second": 47.749,
"train_steps_per_second": 1.492
}
],
"max_steps": 28125,
"num_train_epochs": 3,
"total_flos": 9.03625120512e+16,
"trial_name": null,
"trial_params": null
}