| { | |
| "best_metric": 0.4926412105560303, | |
| "best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l57-l/checkpoint-19500", | |
| "epoch": 2.079778156996587, | |
| "eval_steps": 500, | |
| "global_step": 19500, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.05332764505119454, | |
| "grad_norm": 414.2552795410156, | |
| "learning_rate": 4.911120591581342e-07, | |
| "loss": 0.3809, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.05332764505119454, | |
| "eval_loss": 1.1109352111816406, | |
| "eval_runtime": 63.6734, | |
| "eval_samples_per_second": 15.501, | |
| "eval_steps_per_second": 1.947, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.10665529010238908, | |
| "grad_norm": 45.880027770996094, | |
| "learning_rate": 4.822241183162685e-07, | |
| "loss": 0.2649, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.10665529010238908, | |
| "eval_loss": 0.9588962197303772, | |
| "eval_runtime": 62.6825, | |
| "eval_samples_per_second": 15.746, | |
| "eval_steps_per_second": 1.978, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.1599829351535836, | |
| "grad_norm": 432.9915466308594, | |
| "learning_rate": 4.733361774744027e-07, | |
| "loss": 0.2104, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.1599829351535836, | |
| "eval_loss": 0.90110844373703, | |
| "eval_runtime": 62.6757, | |
| "eval_samples_per_second": 15.748, | |
| "eval_steps_per_second": 1.978, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.21331058020477817, | |
| "grad_norm": 0.014208819717168808, | |
| "learning_rate": 4.6444823663253695e-07, | |
| "loss": 0.1849, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.21331058020477817, | |
| "eval_loss": 0.8570474982261658, | |
| "eval_runtime": 63.7296, | |
| "eval_samples_per_second": 15.487, | |
| "eval_steps_per_second": 1.946, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.2666382252559727, | |
| "grad_norm": 174.73324584960938, | |
| "learning_rate": 4.5556029579067116e-07, | |
| "loss": 0.2056, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.2666382252559727, | |
| "eval_loss": 0.7895939350128174, | |
| "eval_runtime": 63.8387, | |
| "eval_samples_per_second": 15.461, | |
| "eval_steps_per_second": 1.942, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.3199658703071672, | |
| "grad_norm": 9.852073823424234e-09, | |
| "learning_rate": 4.4667235494880547e-07, | |
| "loss": 0.159, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.3199658703071672, | |
| "eval_loss": 0.7647623419761658, | |
| "eval_runtime": 63.9155, | |
| "eval_samples_per_second": 15.442, | |
| "eval_steps_per_second": 1.94, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.37329351535836175, | |
| "grad_norm": 0.0025215104687958956, | |
| "learning_rate": 4.377844141069397e-07, | |
| "loss": 0.1696, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.37329351535836175, | |
| "eval_loss": 0.7638036012649536, | |
| "eval_runtime": 63.8003, | |
| "eval_samples_per_second": 15.47, | |
| "eval_steps_per_second": 1.944, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.42662116040955633, | |
| "grad_norm": 2.4611830711364746, | |
| "learning_rate": 4.2889647326507393e-07, | |
| "loss": 0.1125, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.42662116040955633, | |
| "eval_loss": 0.7486374378204346, | |
| "eval_runtime": 63.9847, | |
| "eval_samples_per_second": 15.426, | |
| "eval_steps_per_second": 1.938, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.47994880546075086, | |
| "grad_norm": 5.230295658111572, | |
| "learning_rate": 4.2000853242320814e-07, | |
| "loss": 0.1322, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.47994880546075086, | |
| "eval_loss": 0.7251659035682678, | |
| "eval_runtime": 64.4782, | |
| "eval_samples_per_second": 15.308, | |
| "eval_steps_per_second": 1.923, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.5332764505119454, | |
| "grad_norm": 3.821002974291332e-05, | |
| "learning_rate": 4.1112059158134245e-07, | |
| "loss": 0.1633, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.5332764505119454, | |
| "eval_loss": 0.7035414576530457, | |
| "eval_runtime": 62.6186, | |
| "eval_samples_per_second": 15.762, | |
| "eval_steps_per_second": 1.98, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.58660409556314, | |
| "grad_norm": 5.916939244343666e-06, | |
| "learning_rate": 4.0223265073947665e-07, | |
| "loss": 0.125, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.58660409556314, | |
| "eval_loss": 0.6792380809783936, | |
| "eval_runtime": 65.0551, | |
| "eval_samples_per_second": 15.172, | |
| "eval_steps_per_second": 1.906, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.6399317406143344, | |
| "grad_norm": 92.8386459350586, | |
| "learning_rate": 3.933447098976109e-07, | |
| "loss": 0.1297, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.6399317406143344, | |
| "eval_loss": 0.6549482345581055, | |
| "eval_runtime": 64.0827, | |
| "eval_samples_per_second": 15.402, | |
| "eval_steps_per_second": 1.935, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.693259385665529, | |
| "grad_norm": 0.00014717792510055006, | |
| "learning_rate": 3.8445676905574517e-07, | |
| "loss": 0.1114, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.693259385665529, | |
| "eval_loss": 0.6223254799842834, | |
| "eval_runtime": 64.89, | |
| "eval_samples_per_second": 15.21, | |
| "eval_steps_per_second": 1.911, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.7465870307167235, | |
| "grad_norm": 4.789559397977428e-07, | |
| "learning_rate": 3.755688282138794e-07, | |
| "loss": 0.144, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.7465870307167235, | |
| "eval_loss": 0.6204637289047241, | |
| "eval_runtime": 62.7674, | |
| "eval_samples_per_second": 15.725, | |
| "eval_steps_per_second": 1.976, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.7999146757679181, | |
| "grad_norm": 1.7078508138656616, | |
| "learning_rate": 3.6668088737201363e-07, | |
| "loss": 0.1399, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.7999146757679181, | |
| "eval_loss": 0.6074371933937073, | |
| "eval_runtime": 63.4694, | |
| "eval_samples_per_second": 15.551, | |
| "eval_steps_per_second": 1.954, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.8532423208191127, | |
| "grad_norm": 870.1183471679688, | |
| "learning_rate": 3.5779294653014783e-07, | |
| "loss": 0.141, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.8532423208191127, | |
| "eval_loss": 0.5950366854667664, | |
| "eval_runtime": 62.915, | |
| "eval_samples_per_second": 15.688, | |
| "eval_steps_per_second": 1.971, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.9065699658703071, | |
| "grad_norm": 5.781071013188921e-05, | |
| "learning_rate": 3.4890500568828214e-07, | |
| "loss": 0.1291, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.9065699658703071, | |
| "eval_loss": 0.5800224542617798, | |
| "eval_runtime": 62.3336, | |
| "eval_samples_per_second": 15.834, | |
| "eval_steps_per_second": 1.989, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.9598976109215017, | |
| "grad_norm": 1.3669992685317993, | |
| "learning_rate": 3.4001706484641635e-07, | |
| "loss": 0.1246, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.9598976109215017, | |
| "eval_loss": 0.5702486038208008, | |
| "eval_runtime": 64.055, | |
| "eval_samples_per_second": 15.409, | |
| "eval_steps_per_second": 1.936, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 1.0132252559726962, | |
| "grad_norm": 644.68115234375, | |
| "learning_rate": 3.311291240045506e-07, | |
| "loss": 0.1257, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 1.0132252559726962, | |
| "eval_loss": 0.5807780623435974, | |
| "eval_runtime": 62.5312, | |
| "eval_samples_per_second": 15.784, | |
| "eval_steps_per_second": 1.983, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 1.0665529010238908, | |
| "grad_norm": 27.854568481445312, | |
| "learning_rate": 3.2224118316268486e-07, | |
| "loss": 0.0336, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 1.0665529010238908, | |
| "eval_loss": 0.5713164210319519, | |
| "eval_runtime": 62.5622, | |
| "eval_samples_per_second": 15.776, | |
| "eval_steps_per_second": 1.982, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 1.1198805460750854, | |
| "grad_norm": 0.00045933027286082506, | |
| "learning_rate": 3.133532423208191e-07, | |
| "loss": 0.0663, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 1.1198805460750854, | |
| "eval_loss": 0.5601483583450317, | |
| "eval_runtime": 63.4675, | |
| "eval_samples_per_second": 15.551, | |
| "eval_steps_per_second": 1.954, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 1.17320819112628, | |
| "grad_norm": 0.14330387115478516, | |
| "learning_rate": 3.044653014789533e-07, | |
| "loss": 0.078, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 1.17320819112628, | |
| "eval_loss": 0.5580261945724487, | |
| "eval_runtime": 63.3927, | |
| "eval_samples_per_second": 15.57, | |
| "eval_steps_per_second": 1.956, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 1.2265358361774745, | |
| "grad_norm": 4.066005518388316e-11, | |
| "learning_rate": 2.955773606370876e-07, | |
| "loss": 0.1118, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 1.2265358361774745, | |
| "eval_loss": 0.5491370558738708, | |
| "eval_runtime": 63.7293, | |
| "eval_samples_per_second": 15.487, | |
| "eval_steps_per_second": 1.946, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 1.2798634812286689, | |
| "grad_norm": 7.519358769059181e-05, | |
| "learning_rate": 2.8668941979522184e-07, | |
| "loss": 0.0534, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 1.2798634812286689, | |
| "eval_loss": 0.5478764772415161, | |
| "eval_runtime": 62.7547, | |
| "eval_samples_per_second": 15.728, | |
| "eval_steps_per_second": 1.976, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 1.3331911262798635, | |
| "grad_norm": 8.400617446113756e-08, | |
| "learning_rate": 2.778014789533561e-07, | |
| "loss": 0.0507, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 1.3331911262798635, | |
| "eval_loss": 0.576600968837738, | |
| "eval_runtime": 64.301, | |
| "eval_samples_per_second": 15.35, | |
| "eval_steps_per_second": 1.928, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 1.386518771331058, | |
| "grad_norm": 2.5385464130084356e-12, | |
| "learning_rate": 2.689135381114903e-07, | |
| "loss": 0.0784, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 1.386518771331058, | |
| "eval_loss": 0.5729931592941284, | |
| "eval_runtime": 63.652, | |
| "eval_samples_per_second": 15.506, | |
| "eval_steps_per_second": 1.948, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 1.4398464163822526, | |
| "grad_norm": 0.08409273624420166, | |
| "learning_rate": 2.600255972696246e-07, | |
| "loss": 0.079, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 1.4398464163822526, | |
| "eval_loss": 0.5694039463996887, | |
| "eval_runtime": 62.8247, | |
| "eval_samples_per_second": 15.71, | |
| "eval_steps_per_second": 1.974, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 1.493174061433447, | |
| "grad_norm": 3.7334253022436314e-08, | |
| "learning_rate": 2.511376564277588e-07, | |
| "loss": 0.0507, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 1.493174061433447, | |
| "eval_loss": 0.5713907480239868, | |
| "eval_runtime": 62.3577, | |
| "eval_samples_per_second": 15.828, | |
| "eval_steps_per_second": 1.989, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 1.5465017064846416, | |
| "grad_norm": 1.5219894647598267, | |
| "learning_rate": 2.422497155858931e-07, | |
| "loss": 0.0774, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 1.5465017064846416, | |
| "eval_loss": 0.5674533843994141, | |
| "eval_runtime": 62.3753, | |
| "eval_samples_per_second": 15.824, | |
| "eval_steps_per_second": 1.988, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 1.5998293515358362, | |
| "grad_norm": 879.1193237304688, | |
| "learning_rate": 2.333617747440273e-07, | |
| "loss": 0.0376, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 1.5998293515358362, | |
| "eval_loss": 0.561253547668457, | |
| "eval_runtime": 62.4788, | |
| "eval_samples_per_second": 15.797, | |
| "eval_steps_per_second": 1.985, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 1.6531569965870307, | |
| "grad_norm": 0.0006221308722160757, | |
| "learning_rate": 2.2447383390216154e-07, | |
| "loss": 0.0929, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 1.6531569965870307, | |
| "eval_loss": 0.5395999550819397, | |
| "eval_runtime": 62.5651, | |
| "eval_samples_per_second": 15.776, | |
| "eval_steps_per_second": 1.982, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 1.7064846416382253, | |
| "grad_norm": 37.85109329223633, | |
| "learning_rate": 2.155858930602958e-07, | |
| "loss": 0.0442, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 1.7064846416382253, | |
| "eval_loss": 0.5307884216308594, | |
| "eval_runtime": 62.4681, | |
| "eval_samples_per_second": 15.8, | |
| "eval_steps_per_second": 1.985, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 1.75981228668942, | |
| "grad_norm": 9.600337885773413e-11, | |
| "learning_rate": 2.0669795221843002e-07, | |
| "loss": 0.0593, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 1.75981228668942, | |
| "eval_loss": 0.5230685472488403, | |
| "eval_runtime": 63.8124, | |
| "eval_samples_per_second": 15.467, | |
| "eval_steps_per_second": 1.943, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 1.8131399317406145, | |
| "grad_norm": 9.067448940129619e-13, | |
| "learning_rate": 1.9781001137656428e-07, | |
| "loss": 0.0578, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 1.8131399317406145, | |
| "eval_loss": 0.5200281739234924, | |
| "eval_runtime": 62.3767, | |
| "eval_samples_per_second": 15.823, | |
| "eval_steps_per_second": 1.988, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 1.8664675767918089, | |
| "grad_norm": 0.00843018852174282, | |
| "learning_rate": 1.889220705346985e-07, | |
| "loss": 0.0928, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 1.8664675767918089, | |
| "eval_loss": 0.509168803691864, | |
| "eval_runtime": 62.4108, | |
| "eval_samples_per_second": 15.815, | |
| "eval_steps_per_second": 1.987, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 1.9197952218430034, | |
| "grad_norm": 4.264631314754297e-09, | |
| "learning_rate": 1.8003412969283277e-07, | |
| "loss": 0.0633, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 1.9197952218430034, | |
| "eval_loss": 0.4954444468021393, | |
| "eval_runtime": 63.3639, | |
| "eval_samples_per_second": 15.577, | |
| "eval_steps_per_second": 1.957, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 1.9731228668941978, | |
| "grad_norm": 459.5137939453125, | |
| "learning_rate": 1.7114618885096697e-07, | |
| "loss": 0.067, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 1.9731228668941978, | |
| "eval_loss": 0.49844062328338623, | |
| "eval_runtime": 63.6573, | |
| "eval_samples_per_second": 15.505, | |
| "eval_steps_per_second": 1.948, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 2.0264505119453924, | |
| "grad_norm": 0.27590689063072205, | |
| "learning_rate": 1.6225824800910123e-07, | |
| "loss": 0.0525, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 2.0264505119453924, | |
| "eval_loss": 0.4999145269393921, | |
| "eval_runtime": 63.1727, | |
| "eval_samples_per_second": 15.624, | |
| "eval_steps_per_second": 1.963, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 2.079778156996587, | |
| "grad_norm": 1.349623744317796e-06, | |
| "learning_rate": 1.533703071672355e-07, | |
| "loss": 0.0301, | |
| "step": 19500 | |
| }, | |
| { | |
| "epoch": 2.079778156996587, | |
| "eval_loss": 0.4926412105560303, | |
| "eval_runtime": 63.4421, | |
| "eval_samples_per_second": 15.557, | |
| "eval_steps_per_second": 1.955, | |
| "step": 19500 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 28128, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 7016007366272520.0, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |