|
{ |
|
"best_metric": 1.2383606433868408, |
|
"best_model_checkpoint": "./vit-base-renovation2/checkpoint-25", |
|
"epoch": 4.0, |
|
"eval_steps": 25, |
|
"global_step": 496, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.08, |
|
"grad_norm": 2.575977325439453, |
|
"learning_rate": 0.00019596774193548388, |
|
"loss": 0.3595, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"grad_norm": 0.9806227684020996, |
|
"learning_rate": 0.00019193548387096775, |
|
"loss": 0.273, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"eval_accuracy": 0.6027397260273972, |
|
"eval_loss": 1.2383606433868408, |
|
"eval_runtime": 7.9057, |
|
"eval_samples_per_second": 27.702, |
|
"eval_steps_per_second": 3.542, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"grad_norm": 2.519897699356079, |
|
"learning_rate": 0.00018790322580645164, |
|
"loss": 0.3335, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"grad_norm": 4.2508063316345215, |
|
"learning_rate": 0.00018387096774193548, |
|
"loss": 0.3046, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"grad_norm": 1.2349441051483154, |
|
"learning_rate": 0.00017983870967741935, |
|
"loss": 0.5153, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"eval_accuracy": 0.5844748858447488, |
|
"eval_loss": 1.40598464012146, |
|
"eval_runtime": 8.1033, |
|
"eval_samples_per_second": 27.026, |
|
"eval_steps_per_second": 3.455, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"grad_norm": 2.6730687618255615, |
|
"learning_rate": 0.00017580645161290325, |
|
"loss": 0.3137, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"grad_norm": 4.323659420013428, |
|
"learning_rate": 0.00017177419354838711, |
|
"loss": 0.2792, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"eval_accuracy": 0.593607305936073, |
|
"eval_loss": 1.30255126953125, |
|
"eval_runtime": 7.4143, |
|
"eval_samples_per_second": 29.538, |
|
"eval_steps_per_second": 3.776, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"grad_norm": 1.2822186946868896, |
|
"learning_rate": 0.00016774193548387098, |
|
"loss": 0.3751, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"grad_norm": 3.15390944480896, |
|
"learning_rate": 0.00016370967741935485, |
|
"loss": 0.5094, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"grad_norm": 3.2424371242523193, |
|
"learning_rate": 0.00016008064516129033, |
|
"loss": 0.5516, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"eval_accuracy": 0.6027397260273972, |
|
"eval_loss": 1.3998719453811646, |
|
"eval_runtime": 7.9973, |
|
"eval_samples_per_second": 27.384, |
|
"eval_steps_per_second": 3.501, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"grad_norm": 1.8185967206954956, |
|
"learning_rate": 0.0001560483870967742, |
|
"loss": 0.2684, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"grad_norm": 3.6117029190063477, |
|
"learning_rate": 0.0001520161290322581, |
|
"loss": 0.4247, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"eval_accuracy": 0.5981735159817352, |
|
"eval_loss": 1.2620913982391357, |
|
"eval_runtime": 8.1833, |
|
"eval_samples_per_second": 26.762, |
|
"eval_steps_per_second": 3.422, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 1.05, |
|
"grad_norm": 0.7001488208770752, |
|
"learning_rate": 0.00014798387096774193, |
|
"loss": 0.1693, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"grad_norm": 3.214643716812134, |
|
"learning_rate": 0.0001439516129032258, |
|
"loss": 0.2085, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 1.21, |
|
"grad_norm": 2.9545109272003174, |
|
"learning_rate": 0.00013991935483870967, |
|
"loss": 0.1556, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.21, |
|
"eval_accuracy": 0.5570776255707762, |
|
"eval_loss": 1.5660632848739624, |
|
"eval_runtime": 8.1924, |
|
"eval_samples_per_second": 26.732, |
|
"eval_steps_per_second": 3.418, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.29, |
|
"grad_norm": 3.702380418777466, |
|
"learning_rate": 0.00013588709677419357, |
|
"loss": 0.1262, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.37, |
|
"grad_norm": 2.634195327758789, |
|
"learning_rate": 0.00013185483870967743, |
|
"loss": 0.1458, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"eval_accuracy": 0.634703196347032, |
|
"eval_loss": 1.3458657264709473, |
|
"eval_runtime": 7.2989, |
|
"eval_samples_per_second": 30.005, |
|
"eval_steps_per_second": 3.836, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"grad_norm": 1.4022108316421509, |
|
"learning_rate": 0.0001278225806451613, |
|
"loss": 0.0859, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"grad_norm": 6.311899185180664, |
|
"learning_rate": 0.00012379032258064514, |
|
"loss": 0.0868, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 1.61, |
|
"grad_norm": 1.674466609954834, |
|
"learning_rate": 0.00011975806451612903, |
|
"loss": 0.1595, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.61, |
|
"eval_accuracy": 0.5981735159817352, |
|
"eval_loss": 1.5277855396270752, |
|
"eval_runtime": 8.087, |
|
"eval_samples_per_second": 27.081, |
|
"eval_steps_per_second": 3.462, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.69, |
|
"grad_norm": 1.2903810739517212, |
|
"learning_rate": 0.00011572580645161291, |
|
"loss": 0.1657, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 1.77, |
|
"grad_norm": 1.2086151838302612, |
|
"learning_rate": 0.00011169354838709678, |
|
"loss": 0.1195, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 1.81, |
|
"eval_accuracy": 0.6255707762557078, |
|
"eval_loss": 1.5303133726119995, |
|
"eval_runtime": 7.5329, |
|
"eval_samples_per_second": 29.072, |
|
"eval_steps_per_second": 3.717, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"grad_norm": 7.646343231201172, |
|
"learning_rate": 0.00010766129032258066, |
|
"loss": 0.1381, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 1.94, |
|
"grad_norm": 5.495790481567383, |
|
"learning_rate": 0.00010362903225806453, |
|
"loss": 0.1761, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"grad_norm": 0.2532494366168976, |
|
"learning_rate": 9.95967741935484e-05, |
|
"loss": 0.1507, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"eval_accuracy": 0.5844748858447488, |
|
"eval_loss": 1.7701371908187866, |
|
"eval_runtime": 8.229, |
|
"eval_samples_per_second": 26.613, |
|
"eval_steps_per_second": 3.403, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 2.1, |
|
"grad_norm": 2.001723527908325, |
|
"learning_rate": 9.556451612903226e-05, |
|
"loss": 0.141, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 2.18, |
|
"grad_norm": 0.07203235477209091, |
|
"learning_rate": 9.153225806451613e-05, |
|
"loss": 0.023, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 2.22, |
|
"eval_accuracy": 0.6301369863013698, |
|
"eval_loss": 1.535449743270874, |
|
"eval_runtime": 7.2739, |
|
"eval_samples_per_second": 30.108, |
|
"eval_steps_per_second": 3.849, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"grad_norm": 0.10443231463432312, |
|
"learning_rate": 8.75e-05, |
|
"loss": 0.0434, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 2.34, |
|
"grad_norm": 1.9513520002365112, |
|
"learning_rate": 8.346774193548388e-05, |
|
"loss": 0.0221, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 2.42, |
|
"grad_norm": 4.193704605102539, |
|
"learning_rate": 7.943548387096774e-05, |
|
"loss": 0.028, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 2.42, |
|
"eval_accuracy": 0.6301369863013698, |
|
"eval_loss": 1.6535084247589111, |
|
"eval_runtime": 8.3156, |
|
"eval_samples_per_second": 26.336, |
|
"eval_steps_per_second": 3.367, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 2.5, |
|
"grad_norm": 0.11177629232406616, |
|
"learning_rate": 7.540322580645162e-05, |
|
"loss": 0.0573, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 2.58, |
|
"grad_norm": 0.0510709173977375, |
|
"learning_rate": 7.137096774193549e-05, |
|
"loss": 0.0698, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 2.62, |
|
"eval_accuracy": 0.6438356164383562, |
|
"eval_loss": 1.6771588325500488, |
|
"eval_runtime": 7.5025, |
|
"eval_samples_per_second": 29.19, |
|
"eval_steps_per_second": 3.732, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 2.66, |
|
"grad_norm": 4.947625160217285, |
|
"learning_rate": 6.733870967741935e-05, |
|
"loss": 0.0964, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 2.74, |
|
"grad_norm": 0.07066596299409866, |
|
"learning_rate": 6.330645161290322e-05, |
|
"loss": 0.0792, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 2.82, |
|
"grad_norm": 6.108673572540283, |
|
"learning_rate": 5.9274193548387104e-05, |
|
"loss": 0.0516, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 2.82, |
|
"eval_accuracy": 0.680365296803653, |
|
"eval_loss": 1.4380197525024414, |
|
"eval_runtime": 8.9956, |
|
"eval_samples_per_second": 24.345, |
|
"eval_steps_per_second": 3.113, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 2.9, |
|
"grad_norm": 0.04943337291479111, |
|
"learning_rate": 5.5241935483870966e-05, |
|
"loss": 0.024, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 2.98, |
|
"grad_norm": 0.2530857026576996, |
|
"learning_rate": 5.120967741935484e-05, |
|
"loss": 0.0136, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 3.02, |
|
"eval_accuracy": 0.6484018264840182, |
|
"eval_loss": 1.6561062335968018, |
|
"eval_runtime": 7.9464, |
|
"eval_samples_per_second": 27.56, |
|
"eval_steps_per_second": 3.524, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 3.06, |
|
"grad_norm": 0.07669265568256378, |
|
"learning_rate": 4.7177419354838716e-05, |
|
"loss": 0.0114, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 3.15, |
|
"grad_norm": 0.04155677929520607, |
|
"learning_rate": 4.3145161290322584e-05, |
|
"loss": 0.0109, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 3.23, |
|
"grad_norm": 0.8200215101242065, |
|
"learning_rate": 3.911290322580645e-05, |
|
"loss": 0.0325, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 3.23, |
|
"eval_accuracy": 0.6621004566210046, |
|
"eval_loss": 1.6027804613113403, |
|
"eval_runtime": 8.0216, |
|
"eval_samples_per_second": 27.301, |
|
"eval_steps_per_second": 3.491, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 3.31, |
|
"grad_norm": 0.028287330642342567, |
|
"learning_rate": 3.508064516129033e-05, |
|
"loss": 0.0085, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 3.39, |
|
"grad_norm": 0.05001891031861305, |
|
"learning_rate": 3.1048387096774195e-05, |
|
"loss": 0.0149, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 3.43, |
|
"eval_accuracy": 0.6621004566210046, |
|
"eval_loss": 1.6261166334152222, |
|
"eval_runtime": 7.3858, |
|
"eval_samples_per_second": 29.652, |
|
"eval_steps_per_second": 3.791, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 3.47, |
|
"grad_norm": 0.07340113818645477, |
|
"learning_rate": 2.7016129032258064e-05, |
|
"loss": 0.0104, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 3.55, |
|
"grad_norm": 0.04165619611740112, |
|
"learning_rate": 2.2983870967741935e-05, |
|
"loss": 0.0119, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 3.63, |
|
"grad_norm": 0.032720282673835754, |
|
"learning_rate": 1.8951612903225807e-05, |
|
"loss": 0.0082, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 3.63, |
|
"eval_accuracy": 0.6621004566210046, |
|
"eval_loss": 1.6614654064178467, |
|
"eval_runtime": 7.1739, |
|
"eval_samples_per_second": 30.527, |
|
"eval_steps_per_second": 3.903, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 3.71, |
|
"grad_norm": 0.07252111285924911, |
|
"learning_rate": 1.4919354838709679e-05, |
|
"loss": 0.0079, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 3.79, |
|
"grad_norm": 0.029246076941490173, |
|
"learning_rate": 1.0887096774193549e-05, |
|
"loss": 0.0093, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 3.83, |
|
"eval_accuracy": 0.6529680365296804, |
|
"eval_loss": 1.6878423690795898, |
|
"eval_runtime": 7.9237, |
|
"eval_samples_per_second": 27.639, |
|
"eval_steps_per_second": 3.534, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 3.87, |
|
"grad_norm": 0.029909249395132065, |
|
"learning_rate": 6.854838709677419e-06, |
|
"loss": 0.0087, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 3.95, |
|
"grad_norm": 0.042575616389513016, |
|
"learning_rate": 2.82258064516129e-06, |
|
"loss": 0.0293, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"step": 496, |
|
"total_flos": 6.10974224738132e+17, |
|
"train_loss": 0.14545002643529686, |
|
"train_runtime": 906.411, |
|
"train_samples_per_second": 8.698, |
|
"train_steps_per_second": 0.547 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 496, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 4, |
|
"save_steps": 25, |
|
"total_flos": 6.10974224738132e+17, |
|
"train_batch_size": 16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|