Augusto777's picture
End of training
7757d98 verified
{
"best_metric": 0.7391304347826086,
"best_model_checkpoint": "swinv2-tiny-patch4-window8-256-DMAE-da-colab/checkpoint-138",
"epoch": 38.26086956521739,
"eval_steps": 500,
"global_step": 440,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.8695652173913043,
"grad_norm": 16.909873962402344,
"learning_rate": 9.090909090909091e-06,
"loss": 1.3823,
"step": 10
},
{
"epoch": 0.9565217391304348,
"eval_accuracy": 0.1956521739130435,
"eval_loss": 1.4058117866516113,
"eval_runtime": 0.9924,
"eval_samples_per_second": 46.35,
"eval_steps_per_second": 3.023,
"step": 11
},
{
"epoch": 1.7391304347826086,
"grad_norm": 19.725204467773438,
"learning_rate": 1.8181818181818182e-05,
"loss": 1.3366,
"step": 20
},
{
"epoch": 2.0,
"eval_accuracy": 0.1956521739130435,
"eval_loss": 1.4482489824295044,
"eval_runtime": 0.7643,
"eval_samples_per_second": 60.185,
"eval_steps_per_second": 3.925,
"step": 23
},
{
"epoch": 2.608695652173913,
"grad_norm": 25.69033432006836,
"learning_rate": 2.7272727272727273e-05,
"loss": 1.2352,
"step": 30
},
{
"epoch": 2.9565217391304346,
"eval_accuracy": 0.45652173913043476,
"eval_loss": 1.230912685394287,
"eval_runtime": 0.8016,
"eval_samples_per_second": 57.387,
"eval_steps_per_second": 3.743,
"step": 34
},
{
"epoch": 3.4782608695652173,
"grad_norm": 22.049612045288086,
"learning_rate": 3.6363636363636364e-05,
"loss": 1.1374,
"step": 40
},
{
"epoch": 4.0,
"eval_accuracy": 0.6086956521739131,
"eval_loss": 1.1031465530395508,
"eval_runtime": 0.9885,
"eval_samples_per_second": 46.536,
"eval_steps_per_second": 3.035,
"step": 46
},
{
"epoch": 4.3478260869565215,
"grad_norm": 29.715892791748047,
"learning_rate": 3.93939393939394e-05,
"loss": 1.0344,
"step": 50
},
{
"epoch": 4.956521739130435,
"eval_accuracy": 0.5869565217391305,
"eval_loss": 1.0230214595794678,
"eval_runtime": 0.7523,
"eval_samples_per_second": 61.145,
"eval_steps_per_second": 3.988,
"step": 57
},
{
"epoch": 5.217391304347826,
"grad_norm": 42.69334030151367,
"learning_rate": 3.838383838383839e-05,
"loss": 0.8772,
"step": 60
},
{
"epoch": 6.0,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 0.9114828705787659,
"eval_runtime": 0.772,
"eval_samples_per_second": 59.587,
"eval_steps_per_second": 3.886,
"step": 69
},
{
"epoch": 6.086956521739131,
"grad_norm": 43.39430618286133,
"learning_rate": 3.7373737373737376e-05,
"loss": 0.8105,
"step": 70
},
{
"epoch": 6.956521739130435,
"grad_norm": 46.86561584472656,
"learning_rate": 3.6363636363636364e-05,
"loss": 0.7321,
"step": 80
},
{
"epoch": 6.956521739130435,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 0.8857656717300415,
"eval_runtime": 1.0277,
"eval_samples_per_second": 44.76,
"eval_steps_per_second": 2.919,
"step": 80
},
{
"epoch": 7.826086956521739,
"grad_norm": 42.23720169067383,
"learning_rate": 3.535353535353536e-05,
"loss": 0.6319,
"step": 90
},
{
"epoch": 8.0,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 0.866457998752594,
"eval_runtime": 0.7747,
"eval_samples_per_second": 59.38,
"eval_steps_per_second": 3.873,
"step": 92
},
{
"epoch": 8.695652173913043,
"grad_norm": 42.64775466918945,
"learning_rate": 3.434343434343435e-05,
"loss": 0.6438,
"step": 100
},
{
"epoch": 8.956521739130435,
"eval_accuracy": 0.717391304347826,
"eval_loss": 0.7737635970115662,
"eval_runtime": 0.7744,
"eval_samples_per_second": 59.4,
"eval_steps_per_second": 3.874,
"step": 103
},
{
"epoch": 9.565217391304348,
"grad_norm": 60.89689636230469,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.4714,
"step": 110
},
{
"epoch": 10.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 0.8491604924201965,
"eval_runtime": 1.0319,
"eval_samples_per_second": 44.578,
"eval_steps_per_second": 2.907,
"step": 115
},
{
"epoch": 10.434782608695652,
"grad_norm": 47.14131546020508,
"learning_rate": 3.232323232323232e-05,
"loss": 0.433,
"step": 120
},
{
"epoch": 10.956521739130435,
"eval_accuracy": 0.6956521739130435,
"eval_loss": 0.8385751247406006,
"eval_runtime": 0.7991,
"eval_samples_per_second": 57.561,
"eval_steps_per_second": 3.754,
"step": 126
},
{
"epoch": 11.304347826086957,
"grad_norm": 40.993927001953125,
"learning_rate": 3.131313131313132e-05,
"loss": 0.4793,
"step": 130
},
{
"epoch": 12.0,
"eval_accuracy": 0.7391304347826086,
"eval_loss": 0.939383327960968,
"eval_runtime": 0.7851,
"eval_samples_per_second": 58.594,
"eval_steps_per_second": 3.821,
"step": 138
},
{
"epoch": 12.173913043478262,
"grad_norm": 48.550899505615234,
"learning_rate": 3.0303030303030306e-05,
"loss": 0.4769,
"step": 140
},
{
"epoch": 12.956521739130435,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 0.9470547437667847,
"eval_runtime": 1.0593,
"eval_samples_per_second": 43.423,
"eval_steps_per_second": 2.832,
"step": 149
},
{
"epoch": 13.043478260869565,
"grad_norm": 84.14537811279297,
"learning_rate": 2.9292929292929297e-05,
"loss": 0.4308,
"step": 150
},
{
"epoch": 13.91304347826087,
"grad_norm": 54.65965270996094,
"learning_rate": 2.8282828282828285e-05,
"loss": 0.3872,
"step": 160
},
{
"epoch": 14.0,
"eval_accuracy": 0.6086956521739131,
"eval_loss": 1.1526373624801636,
"eval_runtime": 0.7735,
"eval_samples_per_second": 59.47,
"eval_steps_per_second": 3.878,
"step": 161
},
{
"epoch": 14.782608695652174,
"grad_norm": 63.24810791015625,
"learning_rate": 2.7272727272727273e-05,
"loss": 0.3906,
"step": 170
},
{
"epoch": 14.956521739130435,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 1.057466983795166,
"eval_runtime": 0.7776,
"eval_samples_per_second": 59.156,
"eval_steps_per_second": 3.858,
"step": 172
},
{
"epoch": 15.652173913043478,
"grad_norm": 76.85520935058594,
"learning_rate": 2.6262626262626265e-05,
"loss": 0.3798,
"step": 180
},
{
"epoch": 16.0,
"eval_accuracy": 0.6956521739130435,
"eval_loss": 1.0592772960662842,
"eval_runtime": 0.8046,
"eval_samples_per_second": 57.172,
"eval_steps_per_second": 3.729,
"step": 184
},
{
"epoch": 16.52173913043478,
"grad_norm": 43.57295608520508,
"learning_rate": 2.5252525252525253e-05,
"loss": 0.3377,
"step": 190
},
{
"epoch": 16.956521739130434,
"eval_accuracy": 0.6086956521739131,
"eval_loss": 1.0782535076141357,
"eval_runtime": 0.7945,
"eval_samples_per_second": 57.9,
"eval_steps_per_second": 3.776,
"step": 195
},
{
"epoch": 17.391304347826086,
"grad_norm": 55.173805236816406,
"learning_rate": 2.4242424242424244e-05,
"loss": 0.3919,
"step": 200
},
{
"epoch": 18.0,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 1.1066713333129883,
"eval_runtime": 0.7682,
"eval_samples_per_second": 59.879,
"eval_steps_per_second": 3.905,
"step": 207
},
{
"epoch": 18.26086956521739,
"grad_norm": 44.900856018066406,
"learning_rate": 2.3232323232323232e-05,
"loss": 0.3631,
"step": 210
},
{
"epoch": 18.956521739130434,
"eval_accuracy": 0.6739130434782609,
"eval_loss": 1.1017966270446777,
"eval_runtime": 0.7687,
"eval_samples_per_second": 59.84,
"eval_steps_per_second": 3.903,
"step": 218
},
{
"epoch": 19.130434782608695,
"grad_norm": 45.883216857910156,
"learning_rate": 2.2222222222222227e-05,
"loss": 0.3433,
"step": 220
},
{
"epoch": 20.0,
"grad_norm": 29.03622055053711,
"learning_rate": 2.121212121212121e-05,
"loss": 0.2762,
"step": 230
},
{
"epoch": 20.0,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 1.147935152053833,
"eval_runtime": 1.0114,
"eval_samples_per_second": 45.482,
"eval_steps_per_second": 2.966,
"step": 230
},
{
"epoch": 20.869565217391305,
"grad_norm": 49.36585235595703,
"learning_rate": 2.0202020202020206e-05,
"loss": 0.2935,
"step": 240
},
{
"epoch": 20.956521739130434,
"eval_accuracy": 0.6956521739130435,
"eval_loss": 1.1054612398147583,
"eval_runtime": 0.7595,
"eval_samples_per_second": 60.565,
"eval_steps_per_second": 3.95,
"step": 241
},
{
"epoch": 21.73913043478261,
"grad_norm": 34.57924270629883,
"learning_rate": 1.9191919191919194e-05,
"loss": 0.3029,
"step": 250
},
{
"epoch": 22.0,
"eval_accuracy": 0.6739130434782609,
"eval_loss": 1.120306372642517,
"eval_runtime": 0.7563,
"eval_samples_per_second": 60.82,
"eval_steps_per_second": 3.967,
"step": 253
},
{
"epoch": 22.608695652173914,
"grad_norm": 50.57252502441406,
"learning_rate": 1.8181818181818182e-05,
"loss": 0.2857,
"step": 260
},
{
"epoch": 22.956521739130434,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.2819756269454956,
"eval_runtime": 0.7606,
"eval_samples_per_second": 60.476,
"eval_steps_per_second": 3.944,
"step": 264
},
{
"epoch": 23.47826086956522,
"grad_norm": 32.393211364746094,
"learning_rate": 1.7171717171717173e-05,
"loss": 0.2603,
"step": 270
},
{
"epoch": 24.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.2550427913665771,
"eval_runtime": 0.7542,
"eval_samples_per_second": 60.991,
"eval_steps_per_second": 3.978,
"step": 276
},
{
"epoch": 24.347826086956523,
"grad_norm": 53.81830596923828,
"learning_rate": 1.616161616161616e-05,
"loss": 0.2162,
"step": 280
},
{
"epoch": 24.956521739130434,
"eval_accuracy": 0.6739130434782609,
"eval_loss": 1.1655433177947998,
"eval_runtime": 0.7904,
"eval_samples_per_second": 58.201,
"eval_steps_per_second": 3.796,
"step": 287
},
{
"epoch": 25.217391304347824,
"grad_norm": 61.60272216796875,
"learning_rate": 1.5151515151515153e-05,
"loss": 0.2465,
"step": 290
},
{
"epoch": 26.0,
"eval_accuracy": 0.6739130434782609,
"eval_loss": 1.2510898113250732,
"eval_runtime": 0.796,
"eval_samples_per_second": 57.789,
"eval_steps_per_second": 3.769,
"step": 299
},
{
"epoch": 26.08695652173913,
"grad_norm": 50.68645477294922,
"learning_rate": 1.4141414141414143e-05,
"loss": 0.2503,
"step": 300
},
{
"epoch": 26.956521739130434,
"grad_norm": 62.43329620361328,
"learning_rate": 1.3131313131313132e-05,
"loss": 0.2238,
"step": 310
},
{
"epoch": 26.956521739130434,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.3460850715637207,
"eval_runtime": 1.0048,
"eval_samples_per_second": 45.782,
"eval_steps_per_second": 2.986,
"step": 310
},
{
"epoch": 27.82608695652174,
"grad_norm": 56.05590057373047,
"learning_rate": 1.2121212121212122e-05,
"loss": 0.2271,
"step": 320
},
{
"epoch": 28.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.3471860885620117,
"eval_runtime": 0.7754,
"eval_samples_per_second": 59.324,
"eval_steps_per_second": 3.869,
"step": 322
},
{
"epoch": 28.695652173913043,
"grad_norm": 36.40557098388672,
"learning_rate": 1.1111111111111113e-05,
"loss": 0.2694,
"step": 330
},
{
"epoch": 28.956521739130434,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4501123428344727,
"eval_runtime": 0.746,
"eval_samples_per_second": 61.663,
"eval_steps_per_second": 4.022,
"step": 333
},
{
"epoch": 29.565217391304348,
"grad_norm": 46.565956115722656,
"learning_rate": 1.0101010101010103e-05,
"loss": 0.1903,
"step": 340
},
{
"epoch": 30.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.462881326675415,
"eval_runtime": 1.0409,
"eval_samples_per_second": 44.193,
"eval_steps_per_second": 2.882,
"step": 345
},
{
"epoch": 30.434782608695652,
"grad_norm": 92.46195220947266,
"learning_rate": 9.090909090909091e-06,
"loss": 0.2054,
"step": 350
},
{
"epoch": 30.956521739130434,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4671651124954224,
"eval_runtime": 0.7747,
"eval_samples_per_second": 59.374,
"eval_steps_per_second": 3.872,
"step": 356
},
{
"epoch": 31.304347826086957,
"grad_norm": 44.79200744628906,
"learning_rate": 8.08080808080808e-06,
"loss": 0.199,
"step": 360
},
{
"epoch": 32.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.472485899925232,
"eval_runtime": 0.7531,
"eval_samples_per_second": 61.078,
"eval_steps_per_second": 3.983,
"step": 368
},
{
"epoch": 32.17391304347826,
"grad_norm": 34.2342643737793,
"learning_rate": 7.070707070707071e-06,
"loss": 0.2034,
"step": 370
},
{
"epoch": 32.95652173913044,
"eval_accuracy": 0.6521739130434783,
"eval_loss": 1.4506837129592896,
"eval_runtime": 0.9808,
"eval_samples_per_second": 46.899,
"eval_steps_per_second": 3.059,
"step": 379
},
{
"epoch": 33.04347826086956,
"grad_norm": 28.347145080566406,
"learning_rate": 6.060606060606061e-06,
"loss": 0.1959,
"step": 380
},
{
"epoch": 33.91304347826087,
"grad_norm": 44.34228515625,
"learning_rate": 5.0505050505050515e-06,
"loss": 0.2048,
"step": 390
},
{
"epoch": 34.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4330270290374756,
"eval_runtime": 0.9488,
"eval_samples_per_second": 48.484,
"eval_steps_per_second": 3.162,
"step": 391
},
{
"epoch": 34.78260869565217,
"grad_norm": 38.90620803833008,
"learning_rate": 4.04040404040404e-06,
"loss": 0.1767,
"step": 400
},
{
"epoch": 34.95652173913044,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4638035297393799,
"eval_runtime": 0.7554,
"eval_samples_per_second": 60.894,
"eval_steps_per_second": 3.971,
"step": 402
},
{
"epoch": 35.65217391304348,
"grad_norm": 32.936920166015625,
"learning_rate": 3.0303030303030305e-06,
"loss": 0.1799,
"step": 410
},
{
"epoch": 36.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4231865406036377,
"eval_runtime": 0.7731,
"eval_samples_per_second": 59.501,
"eval_steps_per_second": 3.88,
"step": 414
},
{
"epoch": 36.52173913043478,
"grad_norm": 32.154483795166016,
"learning_rate": 2.02020202020202e-06,
"loss": 0.1903,
"step": 420
},
{
"epoch": 36.95652173913044,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.450764536857605,
"eval_runtime": 1.0455,
"eval_samples_per_second": 43.997,
"eval_steps_per_second": 2.869,
"step": 425
},
{
"epoch": 37.391304347826086,
"grad_norm": 32.06462860107422,
"learning_rate": 1.01010101010101e-06,
"loss": 0.1864,
"step": 430
},
{
"epoch": 38.0,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4460225105285645,
"eval_runtime": 0.78,
"eval_samples_per_second": 58.972,
"eval_steps_per_second": 3.846,
"step": 437
},
{
"epoch": 38.26086956521739,
"grad_norm": 44.23124694824219,
"learning_rate": 0.0,
"loss": 0.1818,
"step": 440
},
{
"epoch": 38.26086956521739,
"eval_accuracy": 0.6304347826086957,
"eval_loss": 1.4456393718719482,
"eval_runtime": 0.9819,
"eval_samples_per_second": 46.85,
"eval_steps_per_second": 3.055,
"step": 440
},
{
"epoch": 38.26086956521739,
"step": 440,
"total_flos": 8.989085534729011e+17,
"train_loss": 0.4470971633087505,
"train_runtime": 894.3938,
"train_samples_per_second": 32.29,
"train_steps_per_second": 0.492
}
],
"logging_steps": 10,
"max_steps": 440,
"num_input_tokens_seen": 0,
"num_train_epochs": 40,
"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.989085534729011e+17,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}