|
{ |
|
"best_metric": 0.42281684279441833, |
|
"best_model_checkpoint": "/data/jcanete/all_results/pawsx/albeto_tiny/epochs_4_bs_16_lr_5e-5/checkpoint-3000", |
|
"epoch": 4.0, |
|
"global_step": 12352, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.1, |
|
"eval_accuracy": 0.5764999985694885, |
|
"eval_loss": 0.6765033006668091, |
|
"eval_runtime": 0.5222, |
|
"eval_samples_per_second": 3830.009, |
|
"eval_steps_per_second": 239.376, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.798413212435233e-05, |
|
"loss": 0.6642, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"eval_accuracy": 0.7055000066757202, |
|
"eval_loss": 0.5674529671669006, |
|
"eval_runtime": 1.3054, |
|
"eval_samples_per_second": 1532.102, |
|
"eval_steps_per_second": 95.756, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"eval_accuracy": 0.737500011920929, |
|
"eval_loss": 0.5434138178825378, |
|
"eval_runtime": 0.5173, |
|
"eval_samples_per_second": 3866.263, |
|
"eval_steps_per_second": 241.641, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.596421632124353e-05, |
|
"loss": 0.5097, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"eval_accuracy": 0.7609999775886536, |
|
"eval_loss": 0.523171603679657, |
|
"eval_runtime": 1.4935, |
|
"eval_samples_per_second": 1339.147, |
|
"eval_steps_per_second": 83.697, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 4.394025259067357e-05, |
|
"loss": 0.4473, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"eval_accuracy": 0.7864999771118164, |
|
"eval_loss": 0.46667101979255676, |
|
"eval_runtime": 0.5272, |
|
"eval_samples_per_second": 3793.714, |
|
"eval_steps_per_second": 237.107, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"eval_accuracy": 0.8009999990463257, |
|
"eval_loss": 0.4371644854545593, |
|
"eval_runtime": 0.5192, |
|
"eval_samples_per_second": 3852.095, |
|
"eval_steps_per_second": 240.756, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 4.191628886010363e-05, |
|
"loss": 0.3934, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"eval_accuracy": 0.7894999980926514, |
|
"eval_loss": 0.4638134837150574, |
|
"eval_runtime": 0.5279, |
|
"eval_samples_per_second": 3788.294, |
|
"eval_steps_per_second": 236.768, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"eval_accuracy": 0.7885000109672546, |
|
"eval_loss": 0.4593009650707245, |
|
"eval_runtime": 0.5306, |
|
"eval_samples_per_second": 3769.222, |
|
"eval_steps_per_second": 235.576, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 3.989637305699482e-05, |
|
"loss": 0.3659, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"eval_accuracy": 0.7960000038146973, |
|
"eval_loss": 0.4648894965648651, |
|
"eval_runtime": 0.5101, |
|
"eval_samples_per_second": 3920.44, |
|
"eval_steps_per_second": 245.027, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 3.787240932642487e-05, |
|
"loss": 0.3408, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"eval_accuracy": 0.809499979019165, |
|
"eval_loss": 0.42281684279441833, |
|
"eval_runtime": 1.3845, |
|
"eval_samples_per_second": 1444.57, |
|
"eval_steps_per_second": 90.286, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"eval_accuracy": 0.8059999942779541, |
|
"eval_loss": 0.4953405261039734, |
|
"eval_runtime": 0.5733, |
|
"eval_samples_per_second": 3488.535, |
|
"eval_steps_per_second": 218.033, |
|
"step": 3300 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"learning_rate": 3.5848445595854926e-05, |
|
"loss": 0.2812, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"eval_accuracy": 0.7994999885559082, |
|
"eval_loss": 0.5381343364715576, |
|
"eval_runtime": 0.5151, |
|
"eval_samples_per_second": 3882.955, |
|
"eval_steps_per_second": 242.685, |
|
"step": 3600 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"eval_accuracy": 0.8054999709129333, |
|
"eval_loss": 0.4457036256790161, |
|
"eval_runtime": 0.5211, |
|
"eval_samples_per_second": 3838.159, |
|
"eval_steps_per_second": 239.885, |
|
"step": 3900 |
|
}, |
|
{ |
|
"epoch": 1.3, |
|
"learning_rate": 3.382448186528497e-05, |
|
"loss": 0.2545, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.36, |
|
"eval_accuracy": 0.8149999976158142, |
|
"eval_loss": 0.4693449139595032, |
|
"eval_runtime": 0.5216, |
|
"eval_samples_per_second": 3834.156, |
|
"eval_steps_per_second": 239.635, |
|
"step": 4200 |
|
}, |
|
{ |
|
"epoch": 1.46, |
|
"learning_rate": 3.180051813471503e-05, |
|
"loss": 0.2588, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 1.46, |
|
"eval_accuracy": 0.8220000267028809, |
|
"eval_loss": 0.48552772402763367, |
|
"eval_runtime": 2.4236, |
|
"eval_samples_per_second": 825.207, |
|
"eval_steps_per_second": 51.575, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 1.55, |
|
"eval_accuracy": 0.8184999823570251, |
|
"eval_loss": 0.48397132754325867, |
|
"eval_runtime": 0.5143, |
|
"eval_samples_per_second": 3888.874, |
|
"eval_steps_per_second": 243.055, |
|
"step": 4800 |
|
}, |
|
{ |
|
"epoch": 1.62, |
|
"learning_rate": 2.9780602331606216e-05, |
|
"loss": 0.2436, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"eval_accuracy": 0.8080000281333923, |
|
"eval_loss": 0.5025840997695923, |
|
"eval_runtime": 0.5261, |
|
"eval_samples_per_second": 3801.816, |
|
"eval_steps_per_second": 237.613, |
|
"step": 5100 |
|
}, |
|
{ |
|
"epoch": 1.75, |
|
"eval_accuracy": 0.8140000104904175, |
|
"eval_loss": 0.4753943681716919, |
|
"eval_runtime": 0.5312, |
|
"eval_samples_per_second": 3765.313, |
|
"eval_steps_per_second": 235.332, |
|
"step": 5400 |
|
}, |
|
{ |
|
"epoch": 1.78, |
|
"learning_rate": 2.7756638601036272e-05, |
|
"loss": 0.2414, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"eval_accuracy": 0.8034999966621399, |
|
"eval_loss": 0.48492932319641113, |
|
"eval_runtime": 0.5256, |
|
"eval_samples_per_second": 3805.048, |
|
"eval_steps_per_second": 237.815, |
|
"step": 5700 |
|
}, |
|
{ |
|
"epoch": 1.94, |
|
"learning_rate": 2.573267487046632e-05, |
|
"loss": 0.2527, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 1.94, |
|
"eval_accuracy": 0.8100000023841858, |
|
"eval_loss": 0.48092204332351685, |
|
"eval_runtime": 0.52, |
|
"eval_samples_per_second": 3846.385, |
|
"eval_steps_per_second": 240.399, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 2.04, |
|
"eval_accuracy": 0.8134999871253967, |
|
"eval_loss": 0.5660321116447449, |
|
"eval_runtime": 0.5164, |
|
"eval_samples_per_second": 3873.13, |
|
"eval_steps_per_second": 242.071, |
|
"step": 6300 |
|
}, |
|
{ |
|
"epoch": 2.1, |
|
"learning_rate": 2.3708711139896374e-05, |
|
"loss": 0.1818, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"eval_accuracy": 0.8180000185966492, |
|
"eval_loss": 0.6352373361587524, |
|
"eval_runtime": 0.5235, |
|
"eval_samples_per_second": 3820.269, |
|
"eval_steps_per_second": 238.767, |
|
"step": 6600 |
|
}, |
|
{ |
|
"epoch": 2.23, |
|
"eval_accuracy": 0.8209999799728394, |
|
"eval_loss": 0.6510393619537354, |
|
"eval_runtime": 0.5162, |
|
"eval_samples_per_second": 3874.656, |
|
"eval_steps_per_second": 242.166, |
|
"step": 6900 |
|
}, |
|
{ |
|
"epoch": 2.27, |
|
"learning_rate": 2.1692843264248704e-05, |
|
"loss": 0.1776, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 2.33, |
|
"eval_accuracy": 0.8065000176429749, |
|
"eval_loss": 0.684609591960907, |
|
"eval_runtime": 0.5072, |
|
"eval_samples_per_second": 3943.283, |
|
"eval_steps_per_second": 246.455, |
|
"step": 7200 |
|
}, |
|
{ |
|
"epoch": 2.43, |
|
"learning_rate": 1.9668879533678756e-05, |
|
"loss": 0.1754, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 2.43, |
|
"eval_accuracy": 0.8100000023841858, |
|
"eval_loss": 0.6463525295257568, |
|
"eval_runtime": 1.4644, |
|
"eval_samples_per_second": 1365.751, |
|
"eval_steps_per_second": 85.359, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 2.53, |
|
"eval_accuracy": 0.8125, |
|
"eval_loss": 0.6290258169174194, |
|
"eval_runtime": 3.6012, |
|
"eval_samples_per_second": 555.376, |
|
"eval_steps_per_second": 34.711, |
|
"step": 7800 |
|
}, |
|
{ |
|
"epoch": 2.59, |
|
"learning_rate": 1.764491580310881e-05, |
|
"loss": 0.1763, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.62, |
|
"eval_accuracy": 0.8144999742507935, |
|
"eval_loss": 0.6613443493843079, |
|
"eval_runtime": 2.9055, |
|
"eval_samples_per_second": 688.341, |
|
"eval_steps_per_second": 43.021, |
|
"step": 8100 |
|
}, |
|
{ |
|
"epoch": 2.72, |
|
"eval_accuracy": 0.8224999904632568, |
|
"eval_loss": 0.6761817336082458, |
|
"eval_runtime": 1.1073, |
|
"eval_samples_per_second": 1806.22, |
|
"eval_steps_per_second": 112.889, |
|
"step": 8400 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"learning_rate": 1.5620952072538862e-05, |
|
"loss": 0.1853, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 2.82, |
|
"eval_accuracy": 0.8119999766349792, |
|
"eval_loss": 0.6429938077926636, |
|
"eval_runtime": 0.5254, |
|
"eval_samples_per_second": 3806.417, |
|
"eval_steps_per_second": 237.901, |
|
"step": 8700 |
|
}, |
|
{ |
|
"epoch": 2.91, |
|
"learning_rate": 1.3596988341968913e-05, |
|
"loss": 0.1804, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 2.91, |
|
"eval_accuracy": 0.8255000114440918, |
|
"eval_loss": 0.6622639298439026, |
|
"eval_runtime": 1.2718, |
|
"eval_samples_per_second": 1572.63, |
|
"eval_steps_per_second": 98.289, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 3.01, |
|
"eval_accuracy": 0.8245000243186951, |
|
"eval_loss": 0.6402567028999329, |
|
"eval_runtime": 0.5322, |
|
"eval_samples_per_second": 3757.904, |
|
"eval_steps_per_second": 234.869, |
|
"step": 9300 |
|
}, |
|
{ |
|
"epoch": 3.08, |
|
"learning_rate": 1.1577072538860104e-05, |
|
"loss": 0.1321, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 3.11, |
|
"eval_accuracy": 0.8184999823570251, |
|
"eval_loss": 0.76594477891922, |
|
"eval_runtime": 1.2705, |
|
"eval_samples_per_second": 1574.241, |
|
"eval_steps_per_second": 98.39, |
|
"step": 9600 |
|
}, |
|
{ |
|
"epoch": 3.21, |
|
"eval_accuracy": 0.8264999985694885, |
|
"eval_loss": 0.7427929043769836, |
|
"eval_runtime": 0.5386, |
|
"eval_samples_per_second": 3713.17, |
|
"eval_steps_per_second": 232.073, |
|
"step": 9900 |
|
}, |
|
{ |
|
"epoch": 3.24, |
|
"learning_rate": 9.553108808290157e-06, |
|
"loss": 0.1135, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 3.3, |
|
"eval_accuracy": 0.8144999742507935, |
|
"eval_loss": 0.7777068018913269, |
|
"eval_runtime": 2.4153, |
|
"eval_samples_per_second": 828.039, |
|
"eval_steps_per_second": 51.752, |
|
"step": 10200 |
|
}, |
|
{ |
|
"epoch": 3.4, |
|
"learning_rate": 7.529145077720208e-06, |
|
"loss": 0.1182, |
|
"step": 10500 |
|
}, |
|
{ |
|
"epoch": 3.4, |
|
"eval_accuracy": 0.8199999928474426, |
|
"eval_loss": 0.7448311448097229, |
|
"eval_runtime": 0.5257, |
|
"eval_samples_per_second": 3804.49, |
|
"eval_steps_per_second": 237.781, |
|
"step": 10500 |
|
}, |
|
{ |
|
"epoch": 3.5, |
|
"eval_accuracy": 0.824999988079071, |
|
"eval_loss": 0.7690622806549072, |
|
"eval_runtime": 0.5391, |
|
"eval_samples_per_second": 3709.781, |
|
"eval_steps_per_second": 231.861, |
|
"step": 10800 |
|
}, |
|
{ |
|
"epoch": 3.56, |
|
"learning_rate": 5.50518134715026e-06, |
|
"loss": 0.1114, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 3.59, |
|
"eval_accuracy": 0.8195000290870667, |
|
"eval_loss": 0.7780025601387024, |
|
"eval_runtime": 0.5331, |
|
"eval_samples_per_second": 3751.575, |
|
"eval_steps_per_second": 234.473, |
|
"step": 11100 |
|
}, |
|
{ |
|
"epoch": 3.69, |
|
"eval_accuracy": 0.8220000267028809, |
|
"eval_loss": 0.7819697260856628, |
|
"eval_runtime": 0.5267, |
|
"eval_samples_per_second": 3797.154, |
|
"eval_steps_per_second": 237.322, |
|
"step": 11400 |
|
}, |
|
{ |
|
"epoch": 3.72, |
|
"learning_rate": 3.4852655440414507e-06, |
|
"loss": 0.0992, |
|
"step": 11500 |
|
}, |
|
{ |
|
"epoch": 3.79, |
|
"eval_accuracy": 0.8215000033378601, |
|
"eval_loss": 0.79557865858078, |
|
"eval_runtime": 0.537, |
|
"eval_samples_per_second": 3724.493, |
|
"eval_steps_per_second": 232.781, |
|
"step": 11700 |
|
}, |
|
{ |
|
"epoch": 3.89, |
|
"learning_rate": 1.4613018134715026e-06, |
|
"loss": 0.1068, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 3.89, |
|
"eval_accuracy": 0.8224999904632568, |
|
"eval_loss": 0.793406069278717, |
|
"eval_runtime": 0.5363, |
|
"eval_samples_per_second": 3729.122, |
|
"eval_steps_per_second": 233.07, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 3.98, |
|
"eval_accuracy": 0.8215000033378601, |
|
"eval_loss": 0.8001092672348022, |
|
"eval_runtime": 0.5353, |
|
"eval_samples_per_second": 3736.045, |
|
"eval_steps_per_second": 233.503, |
|
"step": 12300 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"step": 12352, |
|
"total_flos": 144723222180000.0, |
|
"train_loss": 0.24578420763806358, |
|
"train_runtime": 1895.6753, |
|
"train_samples_per_second": 104.239, |
|
"train_steps_per_second": 6.516 |
|
} |
|
], |
|
"max_steps": 12352, |
|
"num_train_epochs": 4, |
|
"total_flos": 144723222180000.0, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|