| { | |
| "best_metric": 0.8732149076976663, | |
| "best_model_checkpoint": "vit-base-patch16-224-in21k-finetuned-FER2013/checkpoint-2020", | |
| "epoch": 10.0, | |
| "eval_steps": 500, | |
| "global_step": 2020, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 0.699901819229126, | |
| "learning_rate": 2.4752475247524753e-06, | |
| "loss": 0.7003, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 0.40010973811149597, | |
| "learning_rate": 4.950495049504951e-06, | |
| "loss": 0.6978, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 0.3860187530517578, | |
| "learning_rate": 7.4257425742574256e-06, | |
| "loss": 0.6891, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 0.38579830527305603, | |
| "learning_rate": 9.900990099009901e-06, | |
| "loss": 0.6837, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 0.5694485306739807, | |
| "learning_rate": 1.2376237623762377e-05, | |
| "loss": 0.6643, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "grad_norm": 0.5950794816017151, | |
| "learning_rate": 1.4851485148514851e-05, | |
| "loss": 0.642, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "grad_norm": 0.7590372562408447, | |
| "learning_rate": 1.7326732673267325e-05, | |
| "loss": 0.6213, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "grad_norm": 0.7452055215835571, | |
| "learning_rate": 1.9801980198019803e-05, | |
| "loss": 0.6079, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "grad_norm": 1.1108556985855103, | |
| "learning_rate": 2.227722772277228e-05, | |
| "loss": 0.568, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "grad_norm": 2.0815961360931396, | |
| "learning_rate": 2.4752475247524754e-05, | |
| "loss": 0.5555, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "grad_norm": 2.206498146057129, | |
| "learning_rate": 2.722772277227723e-05, | |
| "loss": 0.5587, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "grad_norm": 0.8447802066802979, | |
| "learning_rate": 2.9702970297029702e-05, | |
| "loss": 0.5502, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "grad_norm": 0.9129859805107117, | |
| "learning_rate": 3.217821782178218e-05, | |
| "loss": 0.5442, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "grad_norm": 1.3158477544784546, | |
| "learning_rate": 3.465346534653465e-05, | |
| "loss": 0.509, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "grad_norm": 1.6786607503890991, | |
| "learning_rate": 3.712871287128713e-05, | |
| "loss": 0.5129, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "grad_norm": 1.0716958045959473, | |
| "learning_rate": 3.9603960396039605e-05, | |
| "loss": 0.5043, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "grad_norm": 1.8416863679885864, | |
| "learning_rate": 4.207920792079208e-05, | |
| "loss": 0.4766, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.89, | |
| "grad_norm": 2.109766721725464, | |
| "learning_rate": 4.455445544554456e-05, | |
| "loss": 0.4746, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "grad_norm": 1.7481091022491455, | |
| "learning_rate": 4.702970297029703e-05, | |
| "loss": 0.5162, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "grad_norm": 1.037062406539917, | |
| "learning_rate": 4.950495049504951e-05, | |
| "loss": 0.4811, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.800417972831766, | |
| "eval_loss": 0.43151769042015076, | |
| "eval_runtime": 15.3075, | |
| "eval_samples_per_second": 187.555, | |
| "eval_steps_per_second": 5.879, | |
| "step": 202 | |
| }, | |
| { | |
| "epoch": 1.04, | |
| "grad_norm": 1.0516308546066284, | |
| "learning_rate": 4.977997799779978e-05, | |
| "loss": 0.4625, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.09, | |
| "grad_norm": 1.0761890411376953, | |
| "learning_rate": 4.950495049504951e-05, | |
| "loss": 0.4659, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.14, | |
| "grad_norm": 1.9692820310592651, | |
| "learning_rate": 4.9229922992299234e-05, | |
| "loss": 0.4893, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "grad_norm": 0.9031820893287659, | |
| "learning_rate": 4.895489548954896e-05, | |
| "loss": 0.4801, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.24, | |
| "grad_norm": 1.241443395614624, | |
| "learning_rate": 4.867986798679868e-05, | |
| "loss": 0.465, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.29, | |
| "grad_norm": 1.1679638624191284, | |
| "learning_rate": 4.8404840484048406e-05, | |
| "loss": 0.4396, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 1.34, | |
| "grad_norm": 1.4820034503936768, | |
| "learning_rate": 4.812981298129813e-05, | |
| "loss": 0.4573, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "grad_norm": 1.0649698972702026, | |
| "learning_rate": 4.785478547854786e-05, | |
| "loss": 0.4616, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 1.44, | |
| "grad_norm": 1.2591346502304077, | |
| "learning_rate": 4.7579757975797585e-05, | |
| "loss": 0.4661, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 1.49, | |
| "grad_norm": 1.9309344291687012, | |
| "learning_rate": 4.730473047304731e-05, | |
| "loss": 0.4792, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.53, | |
| "grad_norm": 0.9727767109870911, | |
| "learning_rate": 4.702970297029703e-05, | |
| "loss": 0.471, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 1.58, | |
| "grad_norm": 1.0227768421173096, | |
| "learning_rate": 4.675467546754676e-05, | |
| "loss": 0.4779, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 1.63, | |
| "grad_norm": 1.7914155721664429, | |
| "learning_rate": 4.647964796479648e-05, | |
| "loss": 0.4552, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 1.68, | |
| "grad_norm": 1.1211057901382446, | |
| "learning_rate": 4.62046204620462e-05, | |
| "loss": 0.4542, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 1.73, | |
| "grad_norm": 0.8994653820991516, | |
| "learning_rate": 4.592959295929593e-05, | |
| "loss": 0.469, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "grad_norm": 1.6254159212112427, | |
| "learning_rate": 4.5654565456545655e-05, | |
| "loss": 0.4209, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 1.83, | |
| "grad_norm": 1.175451397895813, | |
| "learning_rate": 4.537953795379538e-05, | |
| "loss": 0.4707, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 1.88, | |
| "grad_norm": 1.6570440530776978, | |
| "learning_rate": 4.510451045104511e-05, | |
| "loss": 0.4587, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 1.93, | |
| "grad_norm": 1.0916353464126587, | |
| "learning_rate": 4.4829482948294834e-05, | |
| "loss": 0.4507, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 1.98, | |
| "grad_norm": 1.0758109092712402, | |
| "learning_rate": 4.455445544554456e-05, | |
| "loss": 0.4287, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.8432601880877743, | |
| "eval_loss": 0.357921838760376, | |
| "eval_runtime": 15.3987, | |
| "eval_samples_per_second": 186.444, | |
| "eval_steps_per_second": 5.845, | |
| "step": 404 | |
| }, | |
| { | |
| "epoch": 2.03, | |
| "grad_norm": 0.7765569686889648, | |
| "learning_rate": 4.427942794279428e-05, | |
| "loss": 0.4157, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 2.08, | |
| "grad_norm": 1.1473923921585083, | |
| "learning_rate": 4.4004400440044006e-05, | |
| "loss": 0.4227, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 2.13, | |
| "grad_norm": 7.452829837799072, | |
| "learning_rate": 4.372937293729373e-05, | |
| "loss": 0.4033, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 2.18, | |
| "grad_norm": 1.391101360321045, | |
| "learning_rate": 4.345434543454346e-05, | |
| "loss": 0.4563, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 2.23, | |
| "grad_norm": 1.0371557474136353, | |
| "learning_rate": 4.3179317931793185e-05, | |
| "loss": 0.4487, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 2.28, | |
| "grad_norm": 0.998115062713623, | |
| "learning_rate": 4.2904290429042904e-05, | |
| "loss": 0.4178, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "grad_norm": 1.1916583776474, | |
| "learning_rate": 4.262926292629263e-05, | |
| "loss": 0.4361, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 2.38, | |
| "grad_norm": 1.2483221292495728, | |
| "learning_rate": 4.2354235423542356e-05, | |
| "loss": 0.4317, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 2.43, | |
| "grad_norm": 1.2603436708450317, | |
| "learning_rate": 4.207920792079208e-05, | |
| "loss": 0.4252, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 2.48, | |
| "grad_norm": 1.614512324333191, | |
| "learning_rate": 4.18041804180418e-05, | |
| "loss": 0.4213, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 2.52, | |
| "grad_norm": 1.5845041275024414, | |
| "learning_rate": 4.152915291529153e-05, | |
| "loss": 0.4276, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 2.57, | |
| "grad_norm": 0.9629144072532654, | |
| "learning_rate": 4.1254125412541255e-05, | |
| "loss": 0.4168, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 2.62, | |
| "grad_norm": 1.0476891994476318, | |
| "learning_rate": 4.097909790979098e-05, | |
| "loss": 0.4166, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 2.67, | |
| "grad_norm": 1.2000118494033813, | |
| "learning_rate": 4.070407040704071e-05, | |
| "loss": 0.4238, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 2.72, | |
| "grad_norm": 1.608628511428833, | |
| "learning_rate": 4.042904290429043e-05, | |
| "loss": 0.3974, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 2.77, | |
| "grad_norm": 2.0195207595825195, | |
| "learning_rate": 4.015401540154016e-05, | |
| "loss": 0.4087, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 2.82, | |
| "grad_norm": 1.2935571670532227, | |
| "learning_rate": 3.987898789878988e-05, | |
| "loss": 0.396, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 2.87, | |
| "grad_norm": 1.2056093215942383, | |
| "learning_rate": 3.9603960396039605e-05, | |
| "loss": 0.4191, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 2.92, | |
| "grad_norm": 0.9370687007904053, | |
| "learning_rate": 3.932893289328933e-05, | |
| "loss": 0.4323, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "grad_norm": 0.9331934452056885, | |
| "learning_rate": 3.905390539053906e-05, | |
| "loss": 0.4184, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.8467432950191571, | |
| "eval_loss": 0.35166114568710327, | |
| "eval_runtime": 15.3286, | |
| "eval_samples_per_second": 187.297, | |
| "eval_steps_per_second": 5.871, | |
| "step": 606 | |
| }, | |
| { | |
| "epoch": 3.02, | |
| "grad_norm": 1.35804283618927, | |
| "learning_rate": 3.877887788778878e-05, | |
| "loss": 0.4034, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 3.07, | |
| "grad_norm": 1.5728434324264526, | |
| "learning_rate": 3.8503850385038503e-05, | |
| "loss": 0.3917, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 3.12, | |
| "grad_norm": 1.0493104457855225, | |
| "learning_rate": 3.822882288228823e-05, | |
| "loss": 0.416, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 3.17, | |
| "grad_norm": 1.2043938636779785, | |
| "learning_rate": 3.7953795379537956e-05, | |
| "loss": 0.3913, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 3.22, | |
| "grad_norm": 1.6263302564620972, | |
| "learning_rate": 3.767876787678768e-05, | |
| "loss": 0.3815, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 3.27, | |
| "grad_norm": 1.8161449432373047, | |
| "learning_rate": 3.74037403740374e-05, | |
| "loss": 0.3802, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 3.32, | |
| "grad_norm": 1.2450302839279175, | |
| "learning_rate": 3.712871287128713e-05, | |
| "loss": 0.387, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 3.37, | |
| "grad_norm": 1.1230440139770508, | |
| "learning_rate": 3.6853685368536854e-05, | |
| "loss": 0.4023, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 3.42, | |
| "grad_norm": 1.3349334001541138, | |
| "learning_rate": 3.657865786578658e-05, | |
| "loss": 0.4092, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 3.47, | |
| "grad_norm": 1.1348090171813965, | |
| "learning_rate": 3.6303630363036307e-05, | |
| "loss": 0.3724, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 3.51, | |
| "grad_norm": 1.1959893703460693, | |
| "learning_rate": 3.602860286028603e-05, | |
| "loss": 0.415, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 3.56, | |
| "grad_norm": 1.0938156843185425, | |
| "learning_rate": 3.575357535753576e-05, | |
| "loss": 0.3931, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 3.61, | |
| "grad_norm": 1.199876308441162, | |
| "learning_rate": 3.5478547854785485e-05, | |
| "loss": 0.3817, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 3.66, | |
| "grad_norm": 1.3452043533325195, | |
| "learning_rate": 3.5203520352035205e-05, | |
| "loss": 0.3897, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 3.71, | |
| "grad_norm": 1.3791470527648926, | |
| "learning_rate": 3.492849284928493e-05, | |
| "loss": 0.3828, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 3.76, | |
| "grad_norm": 1.7758994102478027, | |
| "learning_rate": 3.465346534653465e-05, | |
| "loss": 0.382, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 3.81, | |
| "grad_norm": 1.0924941301345825, | |
| "learning_rate": 3.4378437843784377e-05, | |
| "loss": 0.4048, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 3.86, | |
| "grad_norm": 1.1729379892349243, | |
| "learning_rate": 3.41034103410341e-05, | |
| "loss": 0.3716, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 3.91, | |
| "grad_norm": 1.1679847240447998, | |
| "learning_rate": 3.382838283828383e-05, | |
| "loss": 0.3726, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 3.96, | |
| "grad_norm": 1.166401743888855, | |
| "learning_rate": 3.3553355335533555e-05, | |
| "loss": 0.3931, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.855451062347614, | |
| "eval_loss": 0.33083683252334595, | |
| "eval_runtime": 15.5188, | |
| "eval_samples_per_second": 185.001, | |
| "eval_steps_per_second": 5.799, | |
| "step": 808 | |
| }, | |
| { | |
| "epoch": 4.01, | |
| "grad_norm": 1.225820541381836, | |
| "learning_rate": 3.327832783278328e-05, | |
| "loss": 0.3976, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 4.06, | |
| "grad_norm": 1.0681790113449097, | |
| "learning_rate": 3.300330033003301e-05, | |
| "loss": 0.3641, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 4.11, | |
| "grad_norm": 1.3162754774093628, | |
| "learning_rate": 3.272827282728273e-05, | |
| "loss": 0.3601, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 4.16, | |
| "grad_norm": 1.5509825944900513, | |
| "learning_rate": 3.2453245324532453e-05, | |
| "loss": 0.3777, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 4.21, | |
| "grad_norm": 1.4965535402297974, | |
| "learning_rate": 3.217821782178218e-05, | |
| "loss": 0.3662, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 4.26, | |
| "grad_norm": 1.846308946609497, | |
| "learning_rate": 3.1903190319031906e-05, | |
| "loss": 0.3755, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 4.31, | |
| "grad_norm": 1.008911371231079, | |
| "learning_rate": 3.162816281628163e-05, | |
| "loss": 0.3283, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 4.36, | |
| "grad_norm": 1.7026519775390625, | |
| "learning_rate": 3.135313531353136e-05, | |
| "loss": 0.3782, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 4.41, | |
| "grad_norm": 1.3682829141616821, | |
| "learning_rate": 3.1078107810781085e-05, | |
| "loss": 0.3943, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 4.46, | |
| "grad_norm": 1.2787247896194458, | |
| "learning_rate": 3.0803080308030804e-05, | |
| "loss": 0.3651, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 4.5, | |
| "grad_norm": 1.2230846881866455, | |
| "learning_rate": 3.052805280528053e-05, | |
| "loss": 0.3489, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 4.55, | |
| "grad_norm": 1.4894435405731201, | |
| "learning_rate": 3.0253025302530253e-05, | |
| "loss": 0.372, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 4.6, | |
| "grad_norm": 1.6204265356063843, | |
| "learning_rate": 2.9977997799779976e-05, | |
| "loss": 0.3637, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 4.65, | |
| "grad_norm": 1.4866546392440796, | |
| "learning_rate": 2.9702970297029702e-05, | |
| "loss": 0.3657, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 4.7, | |
| "grad_norm": 1.8092739582061768, | |
| "learning_rate": 2.942794279427943e-05, | |
| "loss": 0.375, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 4.75, | |
| "grad_norm": 1.3936327695846558, | |
| "learning_rate": 2.9152915291529155e-05, | |
| "loss": 0.3737, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 4.8, | |
| "grad_norm": 1.3176568746566772, | |
| "learning_rate": 2.8877887788778878e-05, | |
| "loss": 0.3565, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 4.85, | |
| "grad_norm": 1.4575417041778564, | |
| "learning_rate": 2.8602860286028604e-05, | |
| "loss": 0.3611, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 4.9, | |
| "grad_norm": 1.5857540369033813, | |
| "learning_rate": 2.832783278327833e-05, | |
| "loss": 0.3656, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 4.95, | |
| "grad_norm": 1.0636128187179565, | |
| "learning_rate": 2.8052805280528056e-05, | |
| "loss": 0.3396, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "grad_norm": 1.4022269248962402, | |
| "learning_rate": 2.777777777777778e-05, | |
| "loss": 0.3667, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.8610240334378265, | |
| "eval_loss": 0.3203551471233368, | |
| "eval_runtime": 15.3956, | |
| "eval_samples_per_second": 186.482, | |
| "eval_steps_per_second": 5.846, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 5.05, | |
| "grad_norm": 1.3242789506912231, | |
| "learning_rate": 2.7502750275027505e-05, | |
| "loss": 0.3563, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 5.1, | |
| "grad_norm": 1.8484835624694824, | |
| "learning_rate": 2.722772277227723e-05, | |
| "loss": 0.3295, | |
| "step": 1030 | |
| }, | |
| { | |
| "epoch": 5.15, | |
| "grad_norm": 1.2119253873825073, | |
| "learning_rate": 2.6952695269526958e-05, | |
| "loss": 0.3789, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 5.2, | |
| "grad_norm": 1.6589637994766235, | |
| "learning_rate": 2.667766776677668e-05, | |
| "loss": 0.3379, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 5.25, | |
| "grad_norm": 1.5714747905731201, | |
| "learning_rate": 2.64026402640264e-05, | |
| "loss": 0.3675, | |
| "step": 1060 | |
| }, | |
| { | |
| "epoch": 5.3, | |
| "grad_norm": 1.3527103662490845, | |
| "learning_rate": 2.6127612761276126e-05, | |
| "loss": 0.3573, | |
| "step": 1070 | |
| }, | |
| { | |
| "epoch": 5.35, | |
| "grad_norm": 1.055513620376587, | |
| "learning_rate": 2.5852585258525853e-05, | |
| "loss": 0.3478, | |
| "step": 1080 | |
| }, | |
| { | |
| "epoch": 5.4, | |
| "grad_norm": 1.6614227294921875, | |
| "learning_rate": 2.557755775577558e-05, | |
| "loss": 0.3692, | |
| "step": 1090 | |
| }, | |
| { | |
| "epoch": 5.45, | |
| "grad_norm": 1.5745677947998047, | |
| "learning_rate": 2.53025302530253e-05, | |
| "loss": 0.331, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 5.5, | |
| "grad_norm": 1.4897499084472656, | |
| "learning_rate": 2.5027502750275028e-05, | |
| "loss": 0.352, | |
| "step": 1110 | |
| }, | |
| { | |
| "epoch": 5.54, | |
| "grad_norm": 1.4070786237716675, | |
| "learning_rate": 2.4752475247524754e-05, | |
| "loss": 0.345, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 5.59, | |
| "grad_norm": 2.2010486125946045, | |
| "learning_rate": 2.447744774477448e-05, | |
| "loss": 0.3213, | |
| "step": 1130 | |
| }, | |
| { | |
| "epoch": 5.64, | |
| "grad_norm": 1.4076497554779053, | |
| "learning_rate": 2.4202420242024203e-05, | |
| "loss": 0.3414, | |
| "step": 1140 | |
| }, | |
| { | |
| "epoch": 5.69, | |
| "grad_norm": 1.168713092803955, | |
| "learning_rate": 2.392739273927393e-05, | |
| "loss": 0.3521, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 5.74, | |
| "grad_norm": 1.8131704330444336, | |
| "learning_rate": 2.3652365236523656e-05, | |
| "loss": 0.3576, | |
| "step": 1160 | |
| }, | |
| { | |
| "epoch": 5.79, | |
| "grad_norm": 1.2432703971862793, | |
| "learning_rate": 2.337733773377338e-05, | |
| "loss": 0.3416, | |
| "step": 1170 | |
| }, | |
| { | |
| "epoch": 5.84, | |
| "grad_norm": 2.248685598373413, | |
| "learning_rate": 2.31023102310231e-05, | |
| "loss": 0.3396, | |
| "step": 1180 | |
| }, | |
| { | |
| "epoch": 5.89, | |
| "grad_norm": 1.8308826684951782, | |
| "learning_rate": 2.2827282728272828e-05, | |
| "loss": 0.3466, | |
| "step": 1190 | |
| }, | |
| { | |
| "epoch": 5.94, | |
| "grad_norm": 2.5257883071899414, | |
| "learning_rate": 2.2552255225522554e-05, | |
| "loss": 0.3186, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 5.99, | |
| "grad_norm": 1.7196460962295532, | |
| "learning_rate": 2.227722772277228e-05, | |
| "loss": 0.3545, | |
| "step": 1210 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.8659003831417624, | |
| "eval_loss": 0.31437239050865173, | |
| "eval_runtime": 15.7136, | |
| "eval_samples_per_second": 182.707, | |
| "eval_steps_per_second": 5.728, | |
| "step": 1212 | |
| }, | |
| { | |
| "epoch": 6.04, | |
| "grad_norm": 1.643741250038147, | |
| "learning_rate": 2.2002200220022003e-05, | |
| "loss": 0.3356, | |
| "step": 1220 | |
| }, | |
| { | |
| "epoch": 6.09, | |
| "grad_norm": 1.5010825395584106, | |
| "learning_rate": 2.172717271727173e-05, | |
| "loss": 0.3288, | |
| "step": 1230 | |
| }, | |
| { | |
| "epoch": 6.14, | |
| "grad_norm": 1.2441235780715942, | |
| "learning_rate": 2.1452145214521452e-05, | |
| "loss": 0.3395, | |
| "step": 1240 | |
| }, | |
| { | |
| "epoch": 6.19, | |
| "grad_norm": 1.351552128791809, | |
| "learning_rate": 2.1177117711771178e-05, | |
| "loss": 0.3011, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 6.24, | |
| "grad_norm": 1.3512260913848877, | |
| "learning_rate": 2.09020902090209e-05, | |
| "loss": 0.3416, | |
| "step": 1260 | |
| }, | |
| { | |
| "epoch": 6.29, | |
| "grad_norm": 1.5813019275665283, | |
| "learning_rate": 2.0627062706270627e-05, | |
| "loss": 0.3523, | |
| "step": 1270 | |
| }, | |
| { | |
| "epoch": 6.34, | |
| "grad_norm": 1.5249056816101074, | |
| "learning_rate": 2.0352035203520354e-05, | |
| "loss": 0.3288, | |
| "step": 1280 | |
| }, | |
| { | |
| "epoch": 6.39, | |
| "grad_norm": 1.9175376892089844, | |
| "learning_rate": 2.007700770077008e-05, | |
| "loss": 0.3527, | |
| "step": 1290 | |
| }, | |
| { | |
| "epoch": 6.44, | |
| "grad_norm": 1.5546538829803467, | |
| "learning_rate": 1.9801980198019803e-05, | |
| "loss": 0.3076, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 6.49, | |
| "grad_norm": 1.5470608472824097, | |
| "learning_rate": 1.952695269526953e-05, | |
| "loss": 0.3303, | |
| "step": 1310 | |
| }, | |
| { | |
| "epoch": 6.53, | |
| "grad_norm": 1.3036069869995117, | |
| "learning_rate": 1.9251925192519252e-05, | |
| "loss": 0.3527, | |
| "step": 1320 | |
| }, | |
| { | |
| "epoch": 6.58, | |
| "grad_norm": 1.524294137954712, | |
| "learning_rate": 1.8976897689768978e-05, | |
| "loss": 0.3243, | |
| "step": 1330 | |
| }, | |
| { | |
| "epoch": 6.63, | |
| "grad_norm": 1.5366973876953125, | |
| "learning_rate": 1.87018701870187e-05, | |
| "loss": 0.3299, | |
| "step": 1340 | |
| }, | |
| { | |
| "epoch": 6.68, | |
| "grad_norm": 1.2275673151016235, | |
| "learning_rate": 1.8426842684268427e-05, | |
| "loss": 0.2928, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 6.73, | |
| "grad_norm": 1.4205721616744995, | |
| "learning_rate": 1.8151815181518153e-05, | |
| "loss": 0.3215, | |
| "step": 1360 | |
| }, | |
| { | |
| "epoch": 6.78, | |
| "grad_norm": 1.4984817504882812, | |
| "learning_rate": 1.787678767876788e-05, | |
| "loss": 0.3008, | |
| "step": 1370 | |
| }, | |
| { | |
| "epoch": 6.83, | |
| "grad_norm": 1.6792049407958984, | |
| "learning_rate": 1.7601760176017602e-05, | |
| "loss": 0.3403, | |
| "step": 1380 | |
| }, | |
| { | |
| "epoch": 6.88, | |
| "grad_norm": 1.5373748540878296, | |
| "learning_rate": 1.7326732673267325e-05, | |
| "loss": 0.3353, | |
| "step": 1390 | |
| }, | |
| { | |
| "epoch": 6.93, | |
| "grad_norm": 2.501495361328125, | |
| "learning_rate": 1.705170517051705e-05, | |
| "loss": 0.3219, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 6.98, | |
| "grad_norm": 1.3856743574142456, | |
| "learning_rate": 1.6776677667766778e-05, | |
| "loss": 0.3137, | |
| "step": 1410 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.864158829676071, | |
| "eval_loss": 0.3307534158229828, | |
| "eval_runtime": 15.9606, | |
| "eval_samples_per_second": 179.88, | |
| "eval_steps_per_second": 5.639, | |
| "step": 1414 | |
| }, | |
| { | |
| "epoch": 7.03, | |
| "grad_norm": 1.5321968793869019, | |
| "learning_rate": 1.6501650165016504e-05, | |
| "loss": 0.308, | |
| "step": 1420 | |
| }, | |
| { | |
| "epoch": 7.08, | |
| "grad_norm": 1.4582158327102661, | |
| "learning_rate": 1.6226622662266227e-05, | |
| "loss": 0.3168, | |
| "step": 1430 | |
| }, | |
| { | |
| "epoch": 7.13, | |
| "grad_norm": 1.4425067901611328, | |
| "learning_rate": 1.5951595159515953e-05, | |
| "loss": 0.3213, | |
| "step": 1440 | |
| }, | |
| { | |
| "epoch": 7.18, | |
| "grad_norm": 1.933104395866394, | |
| "learning_rate": 1.567656765676568e-05, | |
| "loss": 0.2978, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 7.23, | |
| "grad_norm": 1.5549027919769287, | |
| "learning_rate": 1.5401540154015402e-05, | |
| "loss": 0.3281, | |
| "step": 1460 | |
| }, | |
| { | |
| "epoch": 7.28, | |
| "grad_norm": 1.4376338720321655, | |
| "learning_rate": 1.5126512651265127e-05, | |
| "loss": 0.2896, | |
| "step": 1470 | |
| }, | |
| { | |
| "epoch": 7.33, | |
| "grad_norm": 1.414781093597412, | |
| "learning_rate": 1.4851485148514851e-05, | |
| "loss": 0.3135, | |
| "step": 1480 | |
| }, | |
| { | |
| "epoch": 7.38, | |
| "grad_norm": 1.3260300159454346, | |
| "learning_rate": 1.4576457645764577e-05, | |
| "loss": 0.2991, | |
| "step": 1490 | |
| }, | |
| { | |
| "epoch": 7.43, | |
| "grad_norm": 1.8632981777191162, | |
| "learning_rate": 1.4301430143014302e-05, | |
| "loss": 0.3135, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 7.48, | |
| "grad_norm": 1.6400821208953857, | |
| "learning_rate": 1.4026402640264028e-05, | |
| "loss": 0.2882, | |
| "step": 1510 | |
| }, | |
| { | |
| "epoch": 7.52, | |
| "grad_norm": 1.8777371644973755, | |
| "learning_rate": 1.3751375137513753e-05, | |
| "loss": 0.3539, | |
| "step": 1520 | |
| }, | |
| { | |
| "epoch": 7.57, | |
| "grad_norm": 1.431818962097168, | |
| "learning_rate": 1.3476347634763479e-05, | |
| "loss": 0.3065, | |
| "step": 1530 | |
| }, | |
| { | |
| "epoch": 7.62, | |
| "grad_norm": 2.002371311187744, | |
| "learning_rate": 1.32013201320132e-05, | |
| "loss": 0.3362, | |
| "step": 1540 | |
| }, | |
| { | |
| "epoch": 7.67, | |
| "grad_norm": 1.3172950744628906, | |
| "learning_rate": 1.2926292629262926e-05, | |
| "loss": 0.3062, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 7.72, | |
| "grad_norm": 2.155853271484375, | |
| "learning_rate": 1.265126512651265e-05, | |
| "loss": 0.3151, | |
| "step": 1560 | |
| }, | |
| { | |
| "epoch": 7.77, | |
| "grad_norm": 1.8984310626983643, | |
| "learning_rate": 1.2376237623762377e-05, | |
| "loss": 0.3488, | |
| "step": 1570 | |
| }, | |
| { | |
| "epoch": 7.82, | |
| "grad_norm": 1.3027973175048828, | |
| "learning_rate": 1.2101210121012102e-05, | |
| "loss": 0.3197, | |
| "step": 1580 | |
| }, | |
| { | |
| "epoch": 7.87, | |
| "grad_norm": 1.9755516052246094, | |
| "learning_rate": 1.1826182618261828e-05, | |
| "loss": 0.3287, | |
| "step": 1590 | |
| }, | |
| { | |
| "epoch": 7.92, | |
| "grad_norm": 1.6631439924240112, | |
| "learning_rate": 1.155115511551155e-05, | |
| "loss": 0.303, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 7.97, | |
| "grad_norm": 1.8642737865447998, | |
| "learning_rate": 1.1276127612761277e-05, | |
| "loss": 0.3178, | |
| "step": 1610 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.8645071403692093, | |
| "eval_loss": 0.32295528054237366, | |
| "eval_runtime": 15.3242, | |
| "eval_samples_per_second": 187.351, | |
| "eval_steps_per_second": 5.873, | |
| "step": 1616 | |
| }, | |
| { | |
| "epoch": 8.02, | |
| "grad_norm": 1.4315767288208008, | |
| "learning_rate": 1.1001100110011001e-05, | |
| "loss": 0.3078, | |
| "step": 1620 | |
| }, | |
| { | |
| "epoch": 8.07, | |
| "grad_norm": 1.6558310985565186, | |
| "learning_rate": 1.0726072607260726e-05, | |
| "loss": 0.2853, | |
| "step": 1630 | |
| }, | |
| { | |
| "epoch": 8.12, | |
| "grad_norm": 1.906076431274414, | |
| "learning_rate": 1.045104510451045e-05, | |
| "loss": 0.2814, | |
| "step": 1640 | |
| }, | |
| { | |
| "epoch": 8.17, | |
| "grad_norm": 2.086242914199829, | |
| "learning_rate": 1.0176017601760177e-05, | |
| "loss": 0.2847, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 8.22, | |
| "grad_norm": 1.793742299079895, | |
| "learning_rate": 9.900990099009901e-06, | |
| "loss": 0.2945, | |
| "step": 1660 | |
| }, | |
| { | |
| "epoch": 8.27, | |
| "grad_norm": 1.7942755222320557, | |
| "learning_rate": 9.625962596259626e-06, | |
| "loss": 0.296, | |
| "step": 1670 | |
| }, | |
| { | |
| "epoch": 8.32, | |
| "grad_norm": 1.7433489561080933, | |
| "learning_rate": 9.35093509350935e-06, | |
| "loss": 0.2885, | |
| "step": 1680 | |
| }, | |
| { | |
| "epoch": 8.37, | |
| "grad_norm": 1.6308104991912842, | |
| "learning_rate": 9.075907590759077e-06, | |
| "loss": 0.3134, | |
| "step": 1690 | |
| }, | |
| { | |
| "epoch": 8.42, | |
| "grad_norm": 1.6005682945251465, | |
| "learning_rate": 8.800880088008801e-06, | |
| "loss": 0.3206, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 8.47, | |
| "grad_norm": 1.563693642616272, | |
| "learning_rate": 8.525852585258526e-06, | |
| "loss": 0.3112, | |
| "step": 1710 | |
| }, | |
| { | |
| "epoch": 8.51, | |
| "grad_norm": 1.796925663948059, | |
| "learning_rate": 8.250825082508252e-06, | |
| "loss": 0.3076, | |
| "step": 1720 | |
| }, | |
| { | |
| "epoch": 8.56, | |
| "grad_norm": 1.5874933004379272, | |
| "learning_rate": 7.975797579757976e-06, | |
| "loss": 0.3197, | |
| "step": 1730 | |
| }, | |
| { | |
| "epoch": 8.61, | |
| "grad_norm": 1.5310964584350586, | |
| "learning_rate": 7.700770077007701e-06, | |
| "loss": 0.3026, | |
| "step": 1740 | |
| }, | |
| { | |
| "epoch": 8.66, | |
| "grad_norm": 1.9441354274749756, | |
| "learning_rate": 7.4257425742574256e-06, | |
| "loss": 0.3051, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 8.71, | |
| "grad_norm": 1.8038005828857422, | |
| "learning_rate": 7.150715071507151e-06, | |
| "loss": 0.3107, | |
| "step": 1760 | |
| }, | |
| { | |
| "epoch": 8.76, | |
| "grad_norm": 4.733364582061768, | |
| "learning_rate": 6.875687568756876e-06, | |
| "loss": 0.2939, | |
| "step": 1770 | |
| }, | |
| { | |
| "epoch": 8.81, | |
| "grad_norm": 1.5854134559631348, | |
| "learning_rate": 6.6006600660066e-06, | |
| "loss": 0.3049, | |
| "step": 1780 | |
| }, | |
| { | |
| "epoch": 8.86, | |
| "grad_norm": 1.7545024156570435, | |
| "learning_rate": 6.325632563256325e-06, | |
| "loss": 0.2936, | |
| "step": 1790 | |
| }, | |
| { | |
| "epoch": 8.91, | |
| "grad_norm": 1.6307064294815063, | |
| "learning_rate": 6.050605060506051e-06, | |
| "loss": 0.2855, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 8.96, | |
| "grad_norm": 1.6208785772323608, | |
| "learning_rate": 5.775577557755775e-06, | |
| "loss": 0.2998, | |
| "step": 1810 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "eval_accuracy": 0.8707767328456983, | |
| "eval_loss": 0.32057538628578186, | |
| "eval_runtime": 15.4893, | |
| "eval_samples_per_second": 185.354, | |
| "eval_steps_per_second": 5.81, | |
| "step": 1818 | |
| }, | |
| { | |
| "epoch": 9.01, | |
| "grad_norm": 1.7324903011322021, | |
| "learning_rate": 5.500550055005501e-06, | |
| "loss": 0.2916, | |
| "step": 1820 | |
| }, | |
| { | |
| "epoch": 9.06, | |
| "grad_norm": 1.4792495965957642, | |
| "learning_rate": 5.225522552255225e-06, | |
| "loss": 0.2771, | |
| "step": 1830 | |
| }, | |
| { | |
| "epoch": 9.11, | |
| "grad_norm": 2.207402229309082, | |
| "learning_rate": 4.950495049504951e-06, | |
| "loss": 0.2859, | |
| "step": 1840 | |
| }, | |
| { | |
| "epoch": 9.16, | |
| "grad_norm": 1.9077599048614502, | |
| "learning_rate": 4.675467546754675e-06, | |
| "loss": 0.2833, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 9.21, | |
| "grad_norm": 1.7705243825912476, | |
| "learning_rate": 4.400440044004401e-06, | |
| "loss": 0.3007, | |
| "step": 1860 | |
| }, | |
| { | |
| "epoch": 9.26, | |
| "grad_norm": 1.74075448513031, | |
| "learning_rate": 4.125412541254126e-06, | |
| "loss": 0.2912, | |
| "step": 1870 | |
| }, | |
| { | |
| "epoch": 9.31, | |
| "grad_norm": 1.8058630228042603, | |
| "learning_rate": 3.8503850385038505e-06, | |
| "loss": 0.297, | |
| "step": 1880 | |
| }, | |
| { | |
| "epoch": 9.36, | |
| "grad_norm": 1.7161786556243896, | |
| "learning_rate": 3.5753575357535755e-06, | |
| "loss": 0.2789, | |
| "step": 1890 | |
| }, | |
| { | |
| "epoch": 9.41, | |
| "grad_norm": 2.8628289699554443, | |
| "learning_rate": 3.3003300330033e-06, | |
| "loss": 0.3018, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 9.46, | |
| "grad_norm": 1.9037891626358032, | |
| "learning_rate": 3.0253025302530254e-06, | |
| "loss": 0.3043, | |
| "step": 1910 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "grad_norm": 1.5443955659866333, | |
| "learning_rate": 2.7502750275027504e-06, | |
| "loss": 0.2813, | |
| "step": 1920 | |
| }, | |
| { | |
| "epoch": 9.55, | |
| "grad_norm": 1.6395975351333618, | |
| "learning_rate": 2.4752475247524753e-06, | |
| "loss": 0.2942, | |
| "step": 1930 | |
| }, | |
| { | |
| "epoch": 9.6, | |
| "grad_norm": 1.4861618280410767, | |
| "learning_rate": 2.2002200220022003e-06, | |
| "loss": 0.2769, | |
| "step": 1940 | |
| }, | |
| { | |
| "epoch": 9.65, | |
| "grad_norm": 1.9940297603607178, | |
| "learning_rate": 1.9251925192519253e-06, | |
| "loss": 0.2833, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 9.7, | |
| "grad_norm": 1.800776720046997, | |
| "learning_rate": 1.65016501650165e-06, | |
| "loss": 0.2948, | |
| "step": 1960 | |
| }, | |
| { | |
| "epoch": 9.75, | |
| "grad_norm": 1.4160261154174805, | |
| "learning_rate": 1.3751375137513752e-06, | |
| "loss": 0.2932, | |
| "step": 1970 | |
| }, | |
| { | |
| "epoch": 9.8, | |
| "grad_norm": 1.8922903537750244, | |
| "learning_rate": 1.1001100110011001e-06, | |
| "loss": 0.3189, | |
| "step": 1980 | |
| }, | |
| { | |
| "epoch": 9.85, | |
| "grad_norm": 1.754544734954834, | |
| "learning_rate": 8.25082508250825e-07, | |
| "loss": 0.2907, | |
| "step": 1990 | |
| }, | |
| { | |
| "epoch": 9.9, | |
| "grad_norm": 1.8808295726776123, | |
| "learning_rate": 5.500550055005501e-07, | |
| "loss": 0.2763, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 9.95, | |
| "grad_norm": 2.0124707221984863, | |
| "learning_rate": 2.7502750275027504e-07, | |
| "loss": 0.2611, | |
| "step": 2010 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "grad_norm": 1.8302078247070312, | |
| "learning_rate": 0.0, | |
| "loss": 0.2773, | |
| "step": 2020 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy": 0.8732149076976663, | |
| "eval_loss": 0.3263641893863678, | |
| "eval_runtime": 15.9034, | |
| "eval_samples_per_second": 180.527, | |
| "eval_steps_per_second": 5.659, | |
| "step": 2020 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "step": 2020, | |
| "total_flos": 2.0021605356722135e+19, | |
| "train_loss": 0.37925267927717454, | |
| "train_runtime": 3266.1275, | |
| "train_samples_per_second": 79.106, | |
| "train_steps_per_second": 0.618 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 2020, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 10, | |
| "save_steps": 500, | |
| "total_flos": 2.0021605356722135e+19, | |
| "train_batch_size": 32, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |