| { | |
| "best_metric": 0.6213592233009708, | |
| "best_model_checkpoint": "videomae-base-finetuned-kinetics-finetuned-conflab-traj-direction-rh-v2/checkpoint-1053", | |
| "epoch": 8.103513770180436, | |
| "eval_steps": 500, | |
| "global_step": 1053, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.00949667616334283, | |
| "grad_norm": 8.752718925476074, | |
| "learning_rate": 4.716981132075472e-06, | |
| "loss": 2.1131, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.01899335232668566, | |
| "grad_norm": 7.10194206237793, | |
| "learning_rate": 9.433962264150944e-06, | |
| "loss": 2.031, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.02849002849002849, | |
| "grad_norm": 8.048168182373047, | |
| "learning_rate": 1.4150943396226415e-05, | |
| "loss": 2.0382, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.03798670465337132, | |
| "grad_norm": 8.191485404968262, | |
| "learning_rate": 1.8867924528301888e-05, | |
| "loss": 1.9895, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.04748338081671415, | |
| "grad_norm": 7.475042819976807, | |
| "learning_rate": 2.358490566037736e-05, | |
| "loss": 1.9052, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.05698005698005698, | |
| "grad_norm": 12.284913063049316, | |
| "learning_rate": 2.830188679245283e-05, | |
| "loss": 1.9474, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.06647673314339982, | |
| "grad_norm": 6.110039234161377, | |
| "learning_rate": 3.30188679245283e-05, | |
| "loss": 1.8688, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.07597340930674264, | |
| "grad_norm": 9.754578590393066, | |
| "learning_rate": 3.7735849056603776e-05, | |
| "loss": 1.9115, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.08547008547008547, | |
| "grad_norm": 7.101430892944336, | |
| "learning_rate": 4.245283018867925e-05, | |
| "loss": 1.9359, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.0949667616334283, | |
| "grad_norm": 5.658616065979004, | |
| "learning_rate": 4.716981132075472e-05, | |
| "loss": 1.9613, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.10446343779677113, | |
| "grad_norm": 4.824208736419678, | |
| "learning_rate": 4.978880675818374e-05, | |
| "loss": 1.9728, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.1120607787274454, | |
| "eval_accuracy": 0.2912621359223301, | |
| "eval_loss": 1.8081567287445068, | |
| "eval_runtime": 37.3417, | |
| "eval_samples_per_second": 5.517, | |
| "eval_steps_per_second": 0.696, | |
| "step": 118 | |
| }, | |
| { | |
| "epoch": 1.0018993352326686, | |
| "grad_norm": 5.8929443359375, | |
| "learning_rate": 4.9260823653643085e-05, | |
| "loss": 1.8685, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.0113960113960114, | |
| "grad_norm": 5.137537956237793, | |
| "learning_rate": 4.8732840549102435e-05, | |
| "loss": 1.8045, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.0208926875593543, | |
| "grad_norm": 8.457473754882812, | |
| "learning_rate": 4.820485744456177e-05, | |
| "loss": 1.7704, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.030389363722697, | |
| "grad_norm": 6.545841693878174, | |
| "learning_rate": 4.767687434002112e-05, | |
| "loss": 1.7769, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.03988603988604, | |
| "grad_norm": 11.467026710510254, | |
| "learning_rate": 4.7148891235480466e-05, | |
| "loss": 1.664, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.0493827160493827, | |
| "grad_norm": 7.759367942810059, | |
| "learning_rate": 4.662090813093981e-05, | |
| "loss": 1.6638, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.0588793922127255, | |
| "grad_norm": 4.53931188583374, | |
| "learning_rate": 4.609292502639916e-05, | |
| "loss": 1.6305, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.0683760683760684, | |
| "grad_norm": 11.055715560913086, | |
| "learning_rate": 4.55649419218585e-05, | |
| "loss": 1.6884, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.0778727445394112, | |
| "grad_norm": 7.864542007446289, | |
| "learning_rate": 4.503695881731785e-05, | |
| "loss": 1.5855, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.087369420702754, | |
| "grad_norm": 10.435840606689453, | |
| "learning_rate": 4.45089757127772e-05, | |
| "loss": 1.4554, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.0968660968660968, | |
| "grad_norm": 9.551925659179688, | |
| "learning_rate": 4.398099260823654e-05, | |
| "loss": 1.6277, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.1063627730294396, | |
| "grad_norm": 8.910355567932129, | |
| "learning_rate": 4.3453009503695884e-05, | |
| "loss": 1.5903, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.1120607787274455, | |
| "eval_accuracy": 0.3786407766990291, | |
| "eval_loss": 1.803798794746399, | |
| "eval_runtime": 30.5748, | |
| "eval_samples_per_second": 6.738, | |
| "eval_steps_per_second": 0.85, | |
| "step": 236 | |
| }, | |
| { | |
| "epoch": 2.003798670465337, | |
| "grad_norm": 5.9250712394714355, | |
| "learning_rate": 4.292502639915523e-05, | |
| "loss": 1.7366, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 2.01329534662868, | |
| "grad_norm": 7.3831400871276855, | |
| "learning_rate": 4.239704329461457e-05, | |
| "loss": 1.4794, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 2.022792022792023, | |
| "grad_norm": 4.820956230163574, | |
| "learning_rate": 4.186906019007392e-05, | |
| "loss": 1.2516, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 2.0322886989553655, | |
| "grad_norm": 10.172005653381348, | |
| "learning_rate": 4.1341077085533265e-05, | |
| "loss": 1.3548, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 2.0417853751187085, | |
| "grad_norm": 14.752180099487305, | |
| "learning_rate": 4.081309398099261e-05, | |
| "loss": 1.2371, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 2.051282051282051, | |
| "grad_norm": 12.396346092224121, | |
| "learning_rate": 4.028511087645195e-05, | |
| "loss": 1.4009, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 2.060778727445394, | |
| "grad_norm": 11.691078186035156, | |
| "learning_rate": 3.97571277719113e-05, | |
| "loss": 1.2658, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.0702754036087367, | |
| "grad_norm": 7.867719650268555, | |
| "learning_rate": 3.9229144667370646e-05, | |
| "loss": 1.3029, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.07977207977208, | |
| "grad_norm": 11.82673454284668, | |
| "learning_rate": 3.870116156282999e-05, | |
| "loss": 1.1971, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.0892687559354224, | |
| "grad_norm": 12.339221954345703, | |
| "learning_rate": 3.817317845828934e-05, | |
| "loss": 1.3162, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 2.0987654320987654, | |
| "grad_norm": 14.504555702209473, | |
| "learning_rate": 3.764519535374868e-05, | |
| "loss": 1.4248, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.1082621082621085, | |
| "grad_norm": 12.908592224121094, | |
| "learning_rate": 3.711721224920803e-05, | |
| "loss": 1.2371, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.1120607787274452, | |
| "eval_accuracy": 0.4563106796116505, | |
| "eval_loss": 1.466732382774353, | |
| "eval_runtime": 28.3624, | |
| "eval_samples_per_second": 7.263, | |
| "eval_steps_per_second": 0.917, | |
| "step": 354 | |
| }, | |
| { | |
| "epoch": 3.005698005698006, | |
| "grad_norm": 10.089714050292969, | |
| "learning_rate": 3.658922914466738e-05, | |
| "loss": 0.9885, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 3.0151946818613484, | |
| "grad_norm": 6.925798416137695, | |
| "learning_rate": 3.6061246040126714e-05, | |
| "loss": 0.8912, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 3.0246913580246915, | |
| "grad_norm": 7.01603889465332, | |
| "learning_rate": 3.5533262935586064e-05, | |
| "loss": 0.78, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 3.034188034188034, | |
| "grad_norm": 7.25798225402832, | |
| "learning_rate": 3.500527983104541e-05, | |
| "loss": 0.6707, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 3.043684710351377, | |
| "grad_norm": 11.455845832824707, | |
| "learning_rate": 3.447729672650475e-05, | |
| "loss": 0.8491, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 3.0531813865147197, | |
| "grad_norm": 13.49953556060791, | |
| "learning_rate": 3.3949313621964095e-05, | |
| "loss": 0.7744, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 3.0626780626780628, | |
| "grad_norm": 19.157188415527344, | |
| "learning_rate": 3.3421330517423445e-05, | |
| "loss": 0.83, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 3.0721747388414054, | |
| "grad_norm": 15.591106414794922, | |
| "learning_rate": 3.289334741288279e-05, | |
| "loss": 0.8751, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 3.0816714150047484, | |
| "grad_norm": 15.52799129486084, | |
| "learning_rate": 3.236536430834213e-05, | |
| "loss": 0.868, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 3.091168091168091, | |
| "grad_norm": 15.363070487976074, | |
| "learning_rate": 3.183738120380148e-05, | |
| "loss": 0.7792, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 3.100664767331434, | |
| "grad_norm": 9.656035423278809, | |
| "learning_rate": 3.130939809926082e-05, | |
| "loss": 0.6143, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 3.1101614434947766, | |
| "grad_norm": 11.785117149353027, | |
| "learning_rate": 3.078141499472017e-05, | |
| "loss": 0.9635, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 3.1120607787274452, | |
| "eval_accuracy": 0.5631067961165048, | |
| "eval_loss": 1.418686032295227, | |
| "eval_runtime": 23.7661, | |
| "eval_samples_per_second": 8.668, | |
| "eval_steps_per_second": 1.094, | |
| "step": 472 | |
| }, | |
| { | |
| "epoch": 4.007597340930674, | |
| "grad_norm": 5.675172805786133, | |
| "learning_rate": 3.0253431890179517e-05, | |
| "loss": 0.5573, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 4.017094017094017, | |
| "grad_norm": 5.014197826385498, | |
| "learning_rate": 2.972544878563886e-05, | |
| "loss": 0.4382, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 4.02659069325736, | |
| "grad_norm": 8.953272819519043, | |
| "learning_rate": 2.9197465681098207e-05, | |
| "loss": 0.639, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 4.036087369420703, | |
| "grad_norm": 10.838029861450195, | |
| "learning_rate": 2.8669482576557548e-05, | |
| "loss": 0.493, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 4.045584045584046, | |
| "grad_norm": 8.463229179382324, | |
| "learning_rate": 2.8141499472016898e-05, | |
| "loss": 0.4703, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 4.055080721747388, | |
| "grad_norm": 8.104616165161133, | |
| "learning_rate": 2.7613516367476245e-05, | |
| "loss": 0.3912, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 4.064577397910731, | |
| "grad_norm": 13.097394943237305, | |
| "learning_rate": 2.7085533262935585e-05, | |
| "loss": 0.4528, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 4.074074074074074, | |
| "grad_norm": 11.679549217224121, | |
| "learning_rate": 2.6557550158394935e-05, | |
| "loss": 0.5598, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 4.083570750237417, | |
| "grad_norm": 17.88136100769043, | |
| "learning_rate": 2.6029567053854276e-05, | |
| "loss": 0.4933, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 4.09306742640076, | |
| "grad_norm": 4.025656700134277, | |
| "learning_rate": 2.5501583949313622e-05, | |
| "loss": 0.4895, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 4.102564102564102, | |
| "grad_norm": 18.150617599487305, | |
| "learning_rate": 2.497360084477297e-05, | |
| "loss": 0.5344, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 4.112060778727446, | |
| "grad_norm": 10.502995491027832, | |
| "learning_rate": 2.4445617740232313e-05, | |
| "loss": 0.3984, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 4.112060778727446, | |
| "eval_accuracy": 0.5631067961165048, | |
| "eval_loss": 1.398964285850525, | |
| "eval_runtime": 20.6856, | |
| "eval_samples_per_second": 9.959, | |
| "eval_steps_per_second": 1.257, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 5.009496676163343, | |
| "grad_norm": 7.37868070602417, | |
| "learning_rate": 2.391763463569166e-05, | |
| "loss": 0.3346, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 5.018993352326686, | |
| "grad_norm": 5.786819934844971, | |
| "learning_rate": 2.3389651531151003e-05, | |
| "loss": 0.2937, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 5.028490028490029, | |
| "grad_norm": 8.739676475524902, | |
| "learning_rate": 2.286166842661035e-05, | |
| "loss": 0.2478, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 5.037986704653371, | |
| "grad_norm": 3.41898512840271, | |
| "learning_rate": 2.2333685322069694e-05, | |
| "loss": 0.2988, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 5.047483380816714, | |
| "grad_norm": 13.979087829589844, | |
| "learning_rate": 2.180570221752904e-05, | |
| "loss": 0.2672, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 5.056980056980057, | |
| "grad_norm": 9.218252182006836, | |
| "learning_rate": 2.1277719112988384e-05, | |
| "loss": 0.2932, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 5.0664767331434, | |
| "grad_norm": 8.248359680175781, | |
| "learning_rate": 2.074973600844773e-05, | |
| "loss": 0.2845, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 5.075973409306743, | |
| "grad_norm": 12.172591209411621, | |
| "learning_rate": 2.0221752903907075e-05, | |
| "loss": 0.2125, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 5.085470085470085, | |
| "grad_norm": 4.901180267333984, | |
| "learning_rate": 1.9693769799366422e-05, | |
| "loss": 0.2387, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 5.094966761633429, | |
| "grad_norm": 3.2035112380981445, | |
| "learning_rate": 1.9165786694825765e-05, | |
| "loss": 0.1137, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 5.104463437796771, | |
| "grad_norm": 1.4330121278762817, | |
| "learning_rate": 1.863780359028511e-05, | |
| "loss": 0.1962, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 5.112060778727446, | |
| "eval_accuracy": 0.5825242718446602, | |
| "eval_loss": 1.4243208169937134, | |
| "eval_runtime": 20.9667, | |
| "eval_samples_per_second": 9.825, | |
| "eval_steps_per_second": 1.24, | |
| "step": 708 | |
| }, | |
| { | |
| "epoch": 6.001899335232668, | |
| "grad_norm": 17.324275970458984, | |
| "learning_rate": 1.810982048574446e-05, | |
| "loss": 0.2794, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 6.011396011396012, | |
| "grad_norm": 13.693547248840332, | |
| "learning_rate": 1.7581837381203803e-05, | |
| "loss": 0.2361, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 6.020892687559354, | |
| "grad_norm": 1.044037938117981, | |
| "learning_rate": 1.7053854276663146e-05, | |
| "loss": 0.0796, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 6.030389363722697, | |
| "grad_norm": 1.7120157480239868, | |
| "learning_rate": 1.6525871172122493e-05, | |
| "loss": 0.1323, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 6.0398860398860394, | |
| "grad_norm": 16.39497184753418, | |
| "learning_rate": 1.5997888067581837e-05, | |
| "loss": 0.1033, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 6.049382716049383, | |
| "grad_norm": 0.9011927247047424, | |
| "learning_rate": 1.5469904963041184e-05, | |
| "loss": 0.082, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 6.0588793922127255, | |
| "grad_norm": 19.976755142211914, | |
| "learning_rate": 1.4941921858500529e-05, | |
| "loss": 0.1568, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 6.068376068376068, | |
| "grad_norm": 1.4492021799087524, | |
| "learning_rate": 1.4413938753959874e-05, | |
| "loss": 0.1059, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 6.077872744539412, | |
| "grad_norm": 10.073105812072754, | |
| "learning_rate": 1.388595564941922e-05, | |
| "loss": 0.0908, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 6.087369420702754, | |
| "grad_norm": 12.8700532913208, | |
| "learning_rate": 1.3357972544878563e-05, | |
| "loss": 0.1215, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 6.096866096866097, | |
| "grad_norm": 0.7424122095108032, | |
| "learning_rate": 1.2829989440337912e-05, | |
| "loss": 0.1036, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 6.106362773029439, | |
| "grad_norm": 4.486748218536377, | |
| "learning_rate": 1.2302006335797255e-05, | |
| "loss": 0.1952, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 6.112060778727446, | |
| "eval_accuracy": 0.6019417475728155, | |
| "eval_loss": 1.557113528251648, | |
| "eval_runtime": 56.6502, | |
| "eval_samples_per_second": 3.636, | |
| "eval_steps_per_second": 0.459, | |
| "step": 826 | |
| }, | |
| { | |
| "epoch": 7.003798670465337, | |
| "grad_norm": 18.093158721923828, | |
| "learning_rate": 1.17740232312566e-05, | |
| "loss": 0.1356, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 7.01329534662868, | |
| "grad_norm": 0.3082960247993469, | |
| "learning_rate": 1.1246040126715946e-05, | |
| "loss": 0.0692, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 7.022792022792022, | |
| "grad_norm": 0.9313774108886719, | |
| "learning_rate": 1.0718057022175291e-05, | |
| "loss": 0.0197, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 7.032288698955366, | |
| "grad_norm": 0.9466642141342163, | |
| "learning_rate": 1.0190073917634636e-05, | |
| "loss": 0.0202, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 7.0417853751187085, | |
| "grad_norm": 7.768035888671875, | |
| "learning_rate": 9.662090813093982e-06, | |
| "loss": 0.0544, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 7.051282051282051, | |
| "grad_norm": 0.7909216284751892, | |
| "learning_rate": 9.134107708553327e-06, | |
| "loss": 0.0614, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 7.060778727445394, | |
| "grad_norm": 0.6131118535995483, | |
| "learning_rate": 8.606124604012672e-06, | |
| "loss": 0.0458, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 7.070275403608737, | |
| "grad_norm": 0.46935775876045227, | |
| "learning_rate": 8.078141499472017e-06, | |
| "loss": 0.0309, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 7.07977207977208, | |
| "grad_norm": 4.931636333465576, | |
| "learning_rate": 7.5501583949313625e-06, | |
| "loss": 0.0386, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 7.089268755935422, | |
| "grad_norm": 0.7869710326194763, | |
| "learning_rate": 7.022175290390708e-06, | |
| "loss": 0.0599, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 7.098765432098766, | |
| "grad_norm": 18.208322525024414, | |
| "learning_rate": 6.494192185850054e-06, | |
| "loss": 0.1256, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 7.1082621082621085, | |
| "grad_norm": 6.396451473236084, | |
| "learning_rate": 5.966209081309398e-06, | |
| "loss": 0.0319, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 7.112060778727446, | |
| "eval_accuracy": 0.587378640776699, | |
| "eval_loss": 1.5844324827194214, | |
| "eval_runtime": 38.1637, | |
| "eval_samples_per_second": 5.398, | |
| "eval_steps_per_second": 0.681, | |
| "step": 944 | |
| }, | |
| { | |
| "epoch": 8.005698005698006, | |
| "grad_norm": 0.25447767972946167, | |
| "learning_rate": 5.438225976768744e-06, | |
| "loss": 0.0605, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 8.015194681861349, | |
| "grad_norm": 3.442044973373413, | |
| "learning_rate": 4.910242872228089e-06, | |
| "loss": 0.0193, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 8.024691358024691, | |
| "grad_norm": 0.15950489044189453, | |
| "learning_rate": 4.382259767687434e-06, | |
| "loss": 0.0443, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 8.034188034188034, | |
| "grad_norm": 0.21153709292411804, | |
| "learning_rate": 3.854276663146779e-06, | |
| "loss": 0.0152, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 8.043684710351377, | |
| "grad_norm": 0.30688220262527466, | |
| "learning_rate": 3.326293558606125e-06, | |
| "loss": 0.0275, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 8.05318138651472, | |
| "grad_norm": 0.7708584070205688, | |
| "learning_rate": 2.79831045406547e-06, | |
| "loss": 0.0195, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 8.062678062678062, | |
| "grad_norm": 1.1436554193496704, | |
| "learning_rate": 2.2703273495248154e-06, | |
| "loss": 0.0098, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 8.072174738841406, | |
| "grad_norm": 1.9318220615386963, | |
| "learning_rate": 1.7423442449841606e-06, | |
| "loss": 0.0597, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 8.081671415004749, | |
| "grad_norm": 3.5386385917663574, | |
| "learning_rate": 1.2143611404435059e-06, | |
| "loss": 0.0213, | |
| "step": 1030 | |
| }, | |
| { | |
| "epoch": 8.091168091168091, | |
| "grad_norm": 0.16293883323669434, | |
| "learning_rate": 6.863780359028511e-07, | |
| "loss": 0.0118, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 8.100664767331434, | |
| "grad_norm": 2.2753586769104004, | |
| "learning_rate": 1.5839493136219642e-07, | |
| "loss": 0.0203, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 8.103513770180436, | |
| "eval_accuracy": 0.6213592233009708, | |
| "eval_loss": 1.579399824142456, | |
| "eval_runtime": 23.3894, | |
| "eval_samples_per_second": 8.807, | |
| "eval_steps_per_second": 1.112, | |
| "step": 1053 | |
| }, | |
| { | |
| "epoch": 8.103513770180436, | |
| "step": 1053, | |
| "total_flos": 1.0477471675667448e+19, | |
| "train_loss": 0.7528787313023864, | |
| "train_runtime": 2130.6056, | |
| "train_samples_per_second": 3.954, | |
| "train_steps_per_second": 0.494 | |
| }, | |
| { | |
| "epoch": 8.103513770180436, | |
| "eval_accuracy": 0.551219512195122, | |
| "eval_loss": 1.7062230110168457, | |
| "eval_runtime": 32.2785, | |
| "eval_samples_per_second": 6.351, | |
| "eval_steps_per_second": 0.805, | |
| "step": 1053 | |
| }, | |
| { | |
| "epoch": 8.103513770180436, | |
| "eval_accuracy": 0.551219512195122, | |
| "eval_loss": 1.7062228918075562, | |
| "eval_runtime": 25.2134, | |
| "eval_samples_per_second": 8.131, | |
| "eval_steps_per_second": 1.031, | |
| "step": 1053 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 1053, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 9223372036854775807, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "EarlyStoppingCallback": { | |
| "args": { | |
| "early_stopping_patience": 3, | |
| "early_stopping_threshold": 0.0 | |
| }, | |
| "attributes": { | |
| "early_stopping_patience_counter": 0 | |
| } | |
| }, | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.0477471675667448e+19, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |