|
{ |
|
"best_metric": 0.9753602975360297, |
|
"best_model_checkpoint": "vit-base-patch16-224-blur_vs_clean/checkpoint-453", |
|
"epoch": 2.9950413223140497, |
|
"global_step": 453, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 1.0869565217391305e-05, |
|
"loss": 0.7653, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 2.173913043478261e-05, |
|
"loss": 0.6297, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 3.260869565217392e-05, |
|
"loss": 0.4462, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.347826086956522e-05, |
|
"loss": 0.2414, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.950859950859951e-05, |
|
"loss": 0.1249, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 4.8280098280098285e-05, |
|
"loss": 0.1293, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 4.705159705159705e-05, |
|
"loss": 0.1083, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 4.582309582309582e-05, |
|
"loss": 0.0917, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 4.4594594594594596e-05, |
|
"loss": 0.1057, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 4.336609336609337e-05, |
|
"loss": 0.0803, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 4.213759213759214e-05, |
|
"loss": 0.0666, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 4.0909090909090915e-05, |
|
"loss": 0.0832, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 3.968058968058968e-05, |
|
"loss": 0.0867, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 3.845208845208845e-05, |
|
"loss": 0.0812, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 3.7223587223587226e-05, |
|
"loss": 0.0539, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy": 0.9595536959553695, |
|
"eval_loss": 0.10777629911899567, |
|
"eval_runtime": 34.5907, |
|
"eval_samples_per_second": 62.184, |
|
"eval_steps_per_second": 1.966, |
|
"step": 151 |
|
}, |
|
{ |
|
"epoch": 1.06, |
|
"learning_rate": 3.5995085995085995e-05, |
|
"loss": 0.0753, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.12, |
|
"learning_rate": 3.476658476658477e-05, |
|
"loss": 0.0619, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.19, |
|
"learning_rate": 3.3538083538083545e-05, |
|
"loss": 0.0605, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 3.2309582309582306e-05, |
|
"loss": 0.0473, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 1.32, |
|
"learning_rate": 3.108108108108108e-05, |
|
"loss": 0.0672, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.39, |
|
"learning_rate": 2.9852579852579853e-05, |
|
"loss": 0.0626, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"learning_rate": 2.8624078624078625e-05, |
|
"loss": 0.0704, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 1.52, |
|
"learning_rate": 2.73955773955774e-05, |
|
"loss": 0.0517, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 1.59, |
|
"learning_rate": 2.616707616707617e-05, |
|
"loss": 0.0494, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"learning_rate": 2.493857493857494e-05, |
|
"loss": 0.0603, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 1.72, |
|
"learning_rate": 2.371007371007371e-05, |
|
"loss": 0.0678, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 1.79, |
|
"learning_rate": 2.2481572481572483e-05, |
|
"loss": 0.0506, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"learning_rate": 2.1253071253071255e-05, |
|
"loss": 0.0556, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 1.92, |
|
"learning_rate": 2.0024570024570026e-05, |
|
"loss": 0.0687, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 1.98, |
|
"learning_rate": 1.8796068796068798e-05, |
|
"loss": 0.0611, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy": 0.9697814969781497, |
|
"eval_loss": 0.08457668125629425, |
|
"eval_runtime": 33.7533, |
|
"eval_samples_per_second": 63.727, |
|
"eval_steps_per_second": 2.015, |
|
"step": 302 |
|
}, |
|
{ |
|
"epoch": 2.05, |
|
"learning_rate": 1.756756756756757e-05, |
|
"loss": 0.0444, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 2.12, |
|
"learning_rate": 1.6339066339066338e-05, |
|
"loss": 0.0417, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 2.18, |
|
"learning_rate": 1.5110565110565111e-05, |
|
"loss": 0.0496, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 2.25, |
|
"learning_rate": 1.3882063882063885e-05, |
|
"loss": 0.045, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 2.31, |
|
"learning_rate": 1.2653562653562653e-05, |
|
"loss": 0.0496, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 2.38, |
|
"learning_rate": 1.1425061425061426e-05, |
|
"loss": 0.0481, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 2.45, |
|
"learning_rate": 1.0196560196560196e-05, |
|
"loss": 0.0329, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 2.51, |
|
"learning_rate": 8.96805896805897e-06, |
|
"loss": 0.0424, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 2.58, |
|
"learning_rate": 7.73955773955774e-06, |
|
"loss": 0.0274, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 2.64, |
|
"learning_rate": 6.511056511056511e-06, |
|
"loss": 0.0394, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 2.71, |
|
"learning_rate": 5.282555282555283e-06, |
|
"loss": 0.0522, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 2.78, |
|
"learning_rate": 4.0540540540540545e-06, |
|
"loss": 0.0331, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 2.84, |
|
"learning_rate": 2.8255528255528258e-06, |
|
"loss": 0.0275, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 2.91, |
|
"learning_rate": 1.5970515970515972e-06, |
|
"loss": 0.0374, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 2.98, |
|
"learning_rate": 3.6855036855036856e-07, |
|
"loss": 0.049, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_accuracy": 0.9753602975360297, |
|
"eval_loss": 0.07140230387449265, |
|
"eval_runtime": 35.4449, |
|
"eval_samples_per_second": 60.686, |
|
"eval_steps_per_second": 1.918, |
|
"step": 453 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"step": 453, |
|
"total_flos": 4.4917456860202107e+18, |
|
"train_loss": 0.10242866058618028, |
|
"train_runtime": 2277.1585, |
|
"train_samples_per_second": 25.492, |
|
"train_steps_per_second": 0.199 |
|
} |
|
], |
|
"max_steps": 453, |
|
"num_train_epochs": 3, |
|
"total_flos": 4.4917456860202107e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|