|
{ |
|
"best_metric": 0.984313725490196, |
|
"best_model_checkpoint": "swin-tiny-patch4-window7-224-uploads-classifier-v2/checkpoint-54", |
|
"epoch": 20.0, |
|
"global_step": 360, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 2.5e-05, |
|
"loss": 1.2482, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy": 0.8823529411764706, |
|
"eval_loss": 0.47807738184928894, |
|
"eval_runtime": 128.4565, |
|
"eval_samples_per_second": 1.985, |
|
"eval_steps_per_second": 0.062, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 1.11, |
|
"learning_rate": 3.888888888888889e-05, |
|
"loss": 0.6777, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 1.67, |
|
"learning_rate": 4.969135802469136e-05, |
|
"loss": 0.3036, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy": 0.9803921568627451, |
|
"eval_loss": 0.093613401055336, |
|
"eval_runtime": 5.0974, |
|
"eval_samples_per_second": 50.026, |
|
"eval_steps_per_second": 1.569, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 2.22, |
|
"learning_rate": 4.814814814814815e-05, |
|
"loss": 0.2274, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 2.78, |
|
"learning_rate": 4.6604938271604945e-05, |
|
"loss": 0.1687, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_accuracy": 0.984313725490196, |
|
"eval_loss": 0.07451473921537399, |
|
"eval_runtime": 5.1553, |
|
"eval_samples_per_second": 49.463, |
|
"eval_steps_per_second": 1.552, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 3.33, |
|
"learning_rate": 4.506172839506173e-05, |
|
"loss": 0.173, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 3.89, |
|
"learning_rate": 4.351851851851852e-05, |
|
"loss": 0.1392, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_accuracy": 0.9725490196078431, |
|
"eval_loss": 0.09800481051206589, |
|
"eval_runtime": 5.0352, |
|
"eval_samples_per_second": 50.643, |
|
"eval_steps_per_second": 1.589, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 4.44, |
|
"learning_rate": 4.197530864197531e-05, |
|
"loss": 0.1266, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"learning_rate": 4.04320987654321e-05, |
|
"loss": 0.14, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"eval_accuracy": 0.9764705882352941, |
|
"eval_loss": 0.07778824865818024, |
|
"eval_runtime": 5.263, |
|
"eval_samples_per_second": 48.451, |
|
"eval_steps_per_second": 1.52, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 5.56, |
|
"learning_rate": 3.888888888888889e-05, |
|
"loss": 0.1186, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"eval_accuracy": 0.9725490196078431, |
|
"eval_loss": 0.08367796242237091, |
|
"eval_runtime": 5.0122, |
|
"eval_samples_per_second": 50.876, |
|
"eval_steps_per_second": 1.596, |
|
"step": 108 |
|
}, |
|
{ |
|
"epoch": 6.11, |
|
"learning_rate": 3.734567901234568e-05, |
|
"loss": 0.0878, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 6.67, |
|
"learning_rate": 3.580246913580247e-05, |
|
"loss": 0.1088, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"eval_accuracy": 0.9803921568627451, |
|
"eval_loss": 0.06448379158973694, |
|
"eval_runtime": 5.145, |
|
"eval_samples_per_second": 49.562, |
|
"eval_steps_per_second": 1.555, |
|
"step": 126 |
|
}, |
|
{ |
|
"epoch": 7.22, |
|
"learning_rate": 3.425925925925926e-05, |
|
"loss": 0.087, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 7.78, |
|
"learning_rate": 3.271604938271605e-05, |
|
"loss": 0.0789, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"eval_accuracy": 0.9764705882352941, |
|
"eval_loss": 0.0674787312746048, |
|
"eval_runtime": 5.0792, |
|
"eval_samples_per_second": 50.204, |
|
"eval_steps_per_second": 1.575, |
|
"step": 144 |
|
}, |
|
{ |
|
"epoch": 8.33, |
|
"learning_rate": 3.117283950617284e-05, |
|
"loss": 0.0685, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 8.89, |
|
"learning_rate": 2.962962962962963e-05, |
|
"loss": 0.0644, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 9.0, |
|
"eval_accuracy": 0.9686274509803922, |
|
"eval_loss": 0.09402623027563095, |
|
"eval_runtime": 5.1415, |
|
"eval_samples_per_second": 49.597, |
|
"eval_steps_per_second": 1.556, |
|
"step": 162 |
|
}, |
|
{ |
|
"epoch": 9.44, |
|
"learning_rate": 2.8086419753086424e-05, |
|
"loss": 0.0731, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"learning_rate": 2.654320987654321e-05, |
|
"loss": 0.0582, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"eval_accuracy": 0.9725490196078431, |
|
"eval_loss": 0.08792443573474884, |
|
"eval_runtime": 5.0234, |
|
"eval_samples_per_second": 50.762, |
|
"eval_steps_per_second": 1.593, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 10.56, |
|
"learning_rate": 2.5e-05, |
|
"loss": 0.0591, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 11.0, |
|
"eval_accuracy": 0.9686274509803922, |
|
"eval_loss": 0.09348892420530319, |
|
"eval_runtime": 4.9609, |
|
"eval_samples_per_second": 51.402, |
|
"eval_steps_per_second": 1.613, |
|
"step": 198 |
|
}, |
|
{ |
|
"epoch": 11.11, |
|
"learning_rate": 2.345679012345679e-05, |
|
"loss": 0.0432, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 11.67, |
|
"learning_rate": 2.1913580246913582e-05, |
|
"loss": 0.0538, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 12.0, |
|
"eval_accuracy": 0.9803921568627451, |
|
"eval_loss": 0.054012518376111984, |
|
"eval_runtime": 4.9341, |
|
"eval_samples_per_second": 51.681, |
|
"eval_steps_per_second": 1.621, |
|
"step": 216 |
|
}, |
|
{ |
|
"epoch": 12.22, |
|
"learning_rate": 2.037037037037037e-05, |
|
"loss": 0.0471, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 12.78, |
|
"learning_rate": 1.8827160493827163e-05, |
|
"loss": 0.0588, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 13.0, |
|
"eval_accuracy": 0.9686274509803922, |
|
"eval_loss": 0.0725012868642807, |
|
"eval_runtime": 5.0522, |
|
"eval_samples_per_second": 50.473, |
|
"eval_steps_per_second": 1.583, |
|
"step": 234 |
|
}, |
|
{ |
|
"epoch": 13.33, |
|
"learning_rate": 1.728395061728395e-05, |
|
"loss": 0.0508, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 13.89, |
|
"learning_rate": 1.574074074074074e-05, |
|
"loss": 0.0538, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 14.0, |
|
"eval_accuracy": 0.9764705882352941, |
|
"eval_loss": 0.0636754184961319, |
|
"eval_runtime": 5.0696, |
|
"eval_samples_per_second": 50.3, |
|
"eval_steps_per_second": 1.578, |
|
"step": 252 |
|
}, |
|
{ |
|
"epoch": 14.44, |
|
"learning_rate": 1.419753086419753e-05, |
|
"loss": 0.056, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 15.0, |
|
"learning_rate": 1.2654320987654323e-05, |
|
"loss": 0.0462, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 15.0, |
|
"eval_accuracy": 0.9725490196078431, |
|
"eval_loss": 0.06937408447265625, |
|
"eval_runtime": 5.0454, |
|
"eval_samples_per_second": 50.541, |
|
"eval_steps_per_second": 1.586, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 15.56, |
|
"learning_rate": 1.1111111111111112e-05, |
|
"loss": 0.0352, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 16.0, |
|
"eval_accuracy": 0.9686274509803922, |
|
"eval_loss": 0.07712557911872864, |
|
"eval_runtime": 5.0258, |
|
"eval_samples_per_second": 50.738, |
|
"eval_steps_per_second": 1.592, |
|
"step": 288 |
|
}, |
|
{ |
|
"epoch": 16.11, |
|
"learning_rate": 9.5679012345679e-06, |
|
"loss": 0.0469, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 16.67, |
|
"learning_rate": 8.02469135802469e-06, |
|
"loss": 0.0536, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 17.0, |
|
"eval_accuracy": 0.9803921568627451, |
|
"eval_loss": 0.06293778121471405, |
|
"eval_runtime": 4.9305, |
|
"eval_samples_per_second": 51.719, |
|
"eval_steps_per_second": 1.623, |
|
"step": 306 |
|
}, |
|
{ |
|
"epoch": 17.22, |
|
"learning_rate": 6.481481481481481e-06, |
|
"loss": 0.0317, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 17.78, |
|
"learning_rate": 4.938271604938272e-06, |
|
"loss": 0.0403, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 18.0, |
|
"eval_accuracy": 0.9686274509803922, |
|
"eval_loss": 0.09334510564804077, |
|
"eval_runtime": 5.0623, |
|
"eval_samples_per_second": 50.373, |
|
"eval_steps_per_second": 1.58, |
|
"step": 324 |
|
}, |
|
{ |
|
"epoch": 18.33, |
|
"learning_rate": 3.3950617283950617e-06, |
|
"loss": 0.036, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 18.89, |
|
"learning_rate": 1.8518518518518519e-06, |
|
"loss": 0.0412, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 19.0, |
|
"eval_accuracy": 0.9725490196078431, |
|
"eval_loss": 0.08482151478528976, |
|
"eval_runtime": 4.9192, |
|
"eval_samples_per_second": 51.838, |
|
"eval_steps_per_second": 1.626, |
|
"step": 342 |
|
}, |
|
{ |
|
"epoch": 19.44, |
|
"learning_rate": 3.08641975308642e-07, |
|
"loss": 0.029, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"learning_rate": 0.0, |
|
"loss": 0.0305, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"eval_accuracy": 0.9725490196078431, |
|
"eval_loss": 0.08195191621780396, |
|
"eval_runtime": 5.0318, |
|
"eval_samples_per_second": 50.677, |
|
"eval_steps_per_second": 1.59, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"step": 360, |
|
"total_flos": 1.1677328181960376e+18, |
|
"train_loss": 0.13230072516534064, |
|
"train_runtime": 1894.9319, |
|
"train_samples_per_second": 24.149, |
|
"train_steps_per_second": 0.19 |
|
} |
|
], |
|
"max_steps": 360, |
|
"num_train_epochs": 20, |
|
"total_flos": 1.1677328181960376e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|