{ "best_global_step": 705, "best_metric": 0.09983178973197937, "best_model_checkpoint": "/content/drive/MyDrive/efficientnet-b4-nails-model/checkpoint-705", "epoch": 5.0, "eval_steps": 500, "global_step": 1175, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0851063829787234, "grad_norm": 3.302157402038574, "learning_rate": 4.919148936170213e-05, "loss": 1.7304580688476563, "step": 20 }, { "epoch": 0.1702127659574468, "grad_norm": 3.1470284461975098, "learning_rate": 4.8340425531914896e-05, "loss": 1.5841476440429687, "step": 40 }, { "epoch": 0.2553191489361702, "grad_norm": 3.53950834274292, "learning_rate": 4.748936170212766e-05, "loss": 1.4375492095947267, "step": 60 }, { "epoch": 0.3404255319148936, "grad_norm": 3.167618989944458, "learning_rate": 4.6638297872340424e-05, "loss": 1.2950922012329102, "step": 80 }, { "epoch": 0.425531914893617, "grad_norm": 3.141378402709961, "learning_rate": 4.5787234042553194e-05, "loss": 1.0708379745483398, "step": 100 }, { "epoch": 0.5106382978723404, "grad_norm": 3.1946516036987305, "learning_rate": 4.493617021276596e-05, "loss": 0.8597142219543457, "step": 120 }, { "epoch": 0.5957446808510638, "grad_norm": 3.6128246784210205, "learning_rate": 4.408510638297872e-05, "loss": 0.7868041515350341, "step": 140 }, { "epoch": 0.6808510638297872, "grad_norm": 5.161598205566406, "learning_rate": 4.3234042553191486e-05, "loss": 0.698690128326416, "step": 160 }, { "epoch": 0.7659574468085106, "grad_norm": 4.21470308303833, "learning_rate": 4.238297872340426e-05, "loss": 0.6200400829315186, "step": 180 }, { "epoch": 0.851063829787234, "grad_norm": 2.677668809890747, "learning_rate": 4.153191489361702e-05, "loss": 0.5314934253692627, "step": 200 }, { "epoch": 0.9361702127659575, "grad_norm": 1.6603856086730957, "learning_rate": 4.068085106382979e-05, "loss": 0.44487762451171875, "step": 220 }, { "epoch": 1.0, "eval_loss": 0.3377927541732788, "eval_runtime": 7.8572, "eval_samples_per_second": 11.582, "eval_steps_per_second": 0.764, "step": 235 }, { "epoch": 1.0212765957446808, "grad_norm": 4.087282657623291, "learning_rate": 3.9829787234042556e-05, "loss": 0.45683655738830564, "step": 240 }, { "epoch": 1.1063829787234043, "grad_norm": 1.5097193717956543, "learning_rate": 3.897872340425532e-05, "loss": 0.3067047357559204, "step": 260 }, { "epoch": 1.1914893617021276, "grad_norm": 3.3649184703826904, "learning_rate": 3.812765957446809e-05, "loss": 0.2815602540969849, "step": 280 }, { "epoch": 1.2765957446808511, "grad_norm": 2.988518238067627, "learning_rate": 3.7276595744680855e-05, "loss": 0.2793550968170166, "step": 300 }, { "epoch": 1.3617021276595744, "grad_norm": 2.940647840499878, "learning_rate": 3.642553191489362e-05, "loss": 0.28877763748168944, "step": 320 }, { "epoch": 1.4468085106382977, "grad_norm": 4.576767444610596, "learning_rate": 3.557446808510638e-05, "loss": 0.22904467582702637, "step": 340 }, { "epoch": 1.5319148936170213, "grad_norm": 4.684735298156738, "learning_rate": 3.4723404255319154e-05, "loss": 0.26979527473449705, "step": 360 }, { "epoch": 1.6170212765957448, "grad_norm": 3.732126235961914, "learning_rate": 3.387234042553192e-05, "loss": 0.2422767162322998, "step": 380 }, { "epoch": 1.702127659574468, "grad_norm": 3.40220046043396, "learning_rate": 3.302127659574468e-05, "loss": 0.24794645309448243, "step": 400 }, { "epoch": 1.7872340425531914, "grad_norm": 3.6664037704467773, "learning_rate": 3.2170212765957446e-05, "loss": 0.2186866044998169, "step": 420 }, { "epoch": 1.872340425531915, "grad_norm": 6.066251277923584, "learning_rate": 3.131914893617021e-05, "loss": 0.24691376686096192, "step": 440 }, { "epoch": 1.9574468085106385, "grad_norm": 4.297804832458496, "learning_rate": 3.046808510638298e-05, "loss": 0.18424748182296752, "step": 460 }, { "epoch": 2.0, "eval_loss": 0.22569075226783752, "eval_runtime": 1.026, "eval_samples_per_second": 88.69, "eval_steps_per_second": 5.848, "step": 470 }, { "epoch": 2.0425531914893615, "grad_norm": 3.1135361194610596, "learning_rate": 2.9617021276595748e-05, "loss": 0.24354701042175292, "step": 480 }, { "epoch": 2.127659574468085, "grad_norm": 1.9686609506607056, "learning_rate": 2.8765957446808512e-05, "loss": 0.17706258296966554, "step": 500 }, { "epoch": 2.2127659574468086, "grad_norm": 0.9864359498023987, "learning_rate": 2.7914893617021276e-05, "loss": 0.10903202295303345, "step": 520 }, { "epoch": 2.297872340425532, "grad_norm": 2.598639965057373, "learning_rate": 2.7063829787234047e-05, "loss": 0.1576436400413513, "step": 540 }, { "epoch": 2.382978723404255, "grad_norm": 4.634835243225098, "learning_rate": 2.621276595744681e-05, "loss": 0.22807636260986328, "step": 560 }, { "epoch": 2.4680851063829787, "grad_norm": 6.271443843841553, "learning_rate": 2.5361702127659575e-05, "loss": 0.13264695405960084, "step": 580 }, { "epoch": 2.5531914893617023, "grad_norm": 4.2818098068237305, "learning_rate": 2.451063829787234e-05, "loss": 0.1266018271446228, "step": 600 }, { "epoch": 2.6382978723404253, "grad_norm": 5.939896583557129, "learning_rate": 2.3659574468085106e-05, "loss": 0.13802480697631836, "step": 620 }, { "epoch": 2.723404255319149, "grad_norm": 3.1643519401550293, "learning_rate": 2.2808510638297874e-05, "loss": 0.09551984667778016, "step": 640 }, { "epoch": 2.8085106382978724, "grad_norm": 0.462363600730896, "learning_rate": 2.195744680851064e-05, "loss": 0.1383751630783081, "step": 660 }, { "epoch": 2.8936170212765955, "grad_norm": 0.4047956168651581, "learning_rate": 2.1106382978723405e-05, "loss": 0.13573260307312013, "step": 680 }, { "epoch": 2.978723404255319, "grad_norm": 6.376397132873535, "learning_rate": 2.0255319148936172e-05, "loss": 0.1275852680206299, "step": 700 }, { "epoch": 3.0, "eval_loss": 0.09983178973197937, "eval_runtime": 1.0418, "eval_samples_per_second": 87.352, "eval_steps_per_second": 5.759, "step": 705 }, { "epoch": 3.0638297872340425, "grad_norm": 0.7334774136543274, "learning_rate": 1.9404255319148936e-05, "loss": 0.0730510413646698, "step": 720 }, { "epoch": 3.148936170212766, "grad_norm": 0.931010901927948, "learning_rate": 1.8553191489361704e-05, "loss": 0.07135066390037537, "step": 740 }, { "epoch": 3.2340425531914896, "grad_norm": 0.6278732419013977, "learning_rate": 1.7702127659574468e-05, "loss": 0.0836083710193634, "step": 760 }, { "epoch": 3.3191489361702127, "grad_norm": 3.368824005126953, "learning_rate": 1.6851063829787235e-05, "loss": 0.08626257181167603, "step": 780 }, { "epoch": 3.404255319148936, "grad_norm": 0.11537643522024155, "learning_rate": 1.6000000000000003e-05, "loss": 0.11819881200790405, "step": 800 }, { "epoch": 3.4893617021276597, "grad_norm": 0.9512754678726196, "learning_rate": 1.5148936170212765e-05, "loss": 0.06863199472427368, "step": 820 }, { "epoch": 3.574468085106383, "grad_norm": 0.27522802352905273, "learning_rate": 1.4297872340425534e-05, "loss": 0.07966882586479188, "step": 840 }, { "epoch": 3.6595744680851063, "grad_norm": 0.0916527658700943, "learning_rate": 1.3446808510638298e-05, "loss": 0.05501900911331177, "step": 860 }, { "epoch": 3.74468085106383, "grad_norm": 1.5424407720565796, "learning_rate": 1.2595744680851065e-05, "loss": 0.07910426259040833, "step": 880 }, { "epoch": 3.829787234042553, "grad_norm": 2.654416799545288, "learning_rate": 1.174468085106383e-05, "loss": 0.08968157172203065, "step": 900 }, { "epoch": 3.9148936170212765, "grad_norm": 5.711701393127441, "learning_rate": 1.0893617021276597e-05, "loss": 0.07239373326301575, "step": 920 }, { "epoch": 4.0, "grad_norm": 6.0758056640625, "learning_rate": 1.0042553191489363e-05, "loss": 0.06431445479393005, "step": 940 }, { "epoch": 4.0, "eval_loss": 0.17178592085838318, "eval_runtime": 1.0527, "eval_samples_per_second": 86.443, "eval_steps_per_second": 5.7, "step": 940 }, { "epoch": 4.085106382978723, "grad_norm": 0.8240408301353455, "learning_rate": 9.191489361702128e-06, "loss": 0.05246338844299316, "step": 960 }, { "epoch": 4.170212765957447, "grad_norm": 2.452493190765381, "learning_rate": 8.340425531914894e-06, "loss": 0.08099527955055237, "step": 980 }, { "epoch": 4.25531914893617, "grad_norm": 2.292018413543701, "learning_rate": 7.489361702127661e-06, "loss": 0.08736226558685303, "step": 1000 }, { "epoch": 4.340425531914893, "grad_norm": 0.20962312817573547, "learning_rate": 6.6382978723404254e-06, "loss": 0.09467198252677918, "step": 1020 }, { "epoch": 4.425531914893617, "grad_norm": 0.47298091650009155, "learning_rate": 5.787234042553192e-06, "loss": 0.033312416076660155, "step": 1040 }, { "epoch": 4.51063829787234, "grad_norm": 0.2106722593307495, "learning_rate": 4.936170212765958e-06, "loss": 0.04081133008003235, "step": 1060 }, { "epoch": 4.595744680851064, "grad_norm": 4.993804454803467, "learning_rate": 4.085106382978723e-06, "loss": 0.09599153995513916, "step": 1080 }, { "epoch": 4.680851063829787, "grad_norm": 0.2829795181751251, "learning_rate": 3.2340425531914895e-06, "loss": 0.05948078632354736, "step": 1100 }, { "epoch": 4.76595744680851, "grad_norm": 4.379271030426025, "learning_rate": 2.3829787234042557e-06, "loss": 0.07318262457847595, "step": 1120 }, { "epoch": 4.851063829787234, "grad_norm": 6.142686367034912, "learning_rate": 1.5319148936170212e-06, "loss": 0.08857622146606445, "step": 1140 }, { "epoch": 4.9361702127659575, "grad_norm": 1.5748469829559326, "learning_rate": 6.808510638297873e-07, "loss": 0.06185862421989441, "step": 1160 }, { "epoch": 5.0, "eval_loss": 0.1980006843805313, "eval_runtime": 1.1408, "eval_samples_per_second": 79.768, "eval_steps_per_second": 5.259, "step": 1175 } ], "logging_steps": 20, "max_steps": 1175, "num_input_tokens_seen": 0, "num_train_epochs": 5, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 8.57125301637024e+17, "train_batch_size": 16, "trial_name": null, "trial_params": null }