Training in progress, step 10000, checkpoint

checkpoint-10000/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:075aa760d6fd5f4c80ba5e73271f675bc98eea9b346a3e3ef66739e9358d001f
+oid sha256:c5bb8c4163b7576150cfc58a380dad707c33d3aa0cb1b652dbbf3c1de02c020a
 size 1711848436

checkpoint-10000/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:58a4bc9ece8d959bb61198b772f95d1475943d58521c42e37284cf83b5e461a9
+oid sha256:712e475f83f94f0239cd9701e0efc2876cd32aa157c5635bde1178806de19734
 size 3424043887

checkpoint-10000/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fd1f9483f6c91649db661b676a4233e1f965a0ca5eb25be5376d69cc5dfe4ae0
+oid sha256:b38e61a20cae3cd3a87965b8b021073e96ffedd13da1a987fe4ed6052404e23f
 size 623

checkpoint-10000/trainer_state.json

@@ -1,7 +1,7 @@
 {
-  "best_metric": 0.9393072724342346,
+  "best_metric": 0.9735796451568604,
   "best_model_checkpoint": "./output/clip-finetuned-csu-p14-336-e4l58-l/checkpoint-10000",
-  "epoch": 1.8422991893883567,
+  "epoch": 1.0665529010238908,
   "eval_steps": 500,
   "global_step": 10000,
   "is_hyper_param_search": false,
@@ -9,308 +9,308 @@
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 0.09211495946941783,
-      "grad_norm": 20.04043960571289,
-      "learning_rate": 4.8848563006632274e-08,
-      "loss": 0.3758,
+      "epoch": 0.05332764505119454,
+      "grad_norm": 217.55360412597656,
+      "learning_rate": 4.933340443686007e-08,
+      "loss": 0.4667,
       "step": 500
     },
     {
-      "epoch": 0.09211495946941783,
-      "eval_loss": 1.4184927940368652,
-      "eval_runtime": 74.3562,
-      "eval_samples_per_second": 16.233,
-      "eval_steps_per_second": 2.031,
+      "epoch": 0.05332764505119454,
+      "eval_loss": 1.4426143169403076,
+      "eval_runtime": 61.9901,
+      "eval_samples_per_second": 15.922,
+      "eval_steps_per_second": 2.0,
       "step": 500
     },
     {
-      "epoch": 0.18422991893883567,
-      "grad_norm": 74.25025939941406,
-      "learning_rate": 4.7697126013264556e-08,
-      "loss": 0.4103,
+      "epoch": 0.10665529010238908,
+      "grad_norm": 73.73340606689453,
+      "learning_rate": 4.8666808873720136e-08,
+      "loss": 0.4532,
       "step": 1000
     },
     {
-      "epoch": 0.18422991893883567,
-      "eval_loss": 1.3500770330429077,
-      "eval_runtime": 75.7013,
-      "eval_samples_per_second": 15.944,
-      "eval_steps_per_second": 1.995,
+      "epoch": 0.10665529010238908,
+      "eval_loss": 1.3815597295761108,
+      "eval_runtime": 62.4624,
+      "eval_samples_per_second": 15.801,
+      "eval_steps_per_second": 1.985,
       "step": 1000
     },
     {
-      "epoch": 0.2763448784082535,
-      "grad_norm": 0.5102410912513733,
-      "learning_rate": 4.6545689019896826e-08,
-      "loss": 0.433,
+      "epoch": 0.1599829351535836,
+      "grad_norm": 471.4011535644531,
+      "learning_rate": 4.80002133105802e-08,
+      "loss": 0.3749,
       "step": 1500
     },
     {
-      "epoch": 0.2763448784082535,
-      "eval_loss": 1.2885024547576904,
-      "eval_runtime": 75.197,
-      "eval_samples_per_second": 16.051,
-      "eval_steps_per_second": 2.008,
+      "epoch": 0.1599829351535836,
+      "eval_loss": 1.3310539722442627,
+      "eval_runtime": 63.3918,
+      "eval_samples_per_second": 15.57,
+      "eval_steps_per_second": 1.956,
       "step": 1500
     },
     {
-      "epoch": 0.36845983787767134,
-      "grad_norm": 0.1667678952217102,
-      "learning_rate": 4.539425202652911e-08,
-      "loss": 0.3424,
+      "epoch": 0.21331058020477817,
+      "grad_norm": 11.962693214416504,
+      "learning_rate": 4.733361774744027e-08,
+      "loss": 0.336,
       "step": 2000
     },
     {
-      "epoch": 0.36845983787767134,
-      "eval_loss": 1.239119052886963,
-      "eval_runtime": 75.387,
-      "eval_samples_per_second": 16.011,
-      "eval_steps_per_second": 2.003,
+      "epoch": 0.21331058020477817,
+      "eval_loss": 1.2890639305114746,
+      "eval_runtime": 63.177,
+      "eval_samples_per_second": 15.623,
+      "eval_steps_per_second": 1.963,
       "step": 2000
     },
     {
-      "epoch": 0.46057479734708917,
-      "grad_norm": 343.8175048828125,
-      "learning_rate": 4.4242815033161385e-08,
-      "loss": 0.3645,
+      "epoch": 0.2666382252559727,
+      "grad_norm": 375.1609191894531,
+      "learning_rate": 4.666702218430034e-08,
+      "loss": 0.3585,
       "step": 2500
     },
     {
-      "epoch": 0.46057479734708917,
-      "eval_loss": 1.1902339458465576,
-      "eval_runtime": 75.9899,
-      "eval_samples_per_second": 15.884,
-      "eval_steps_per_second": 1.987,
+      "epoch": 0.2666382252559727,
+      "eval_loss": 1.2536433935165405,
+      "eval_runtime": 63.1512,
+      "eval_samples_per_second": 15.629,
+      "eval_steps_per_second": 1.964,
       "step": 2500
     },
     {
-      "epoch": 0.552689756816507,
-      "grad_norm": 305.5626525878906,
-      "learning_rate": 4.309137803979366e-08,
-      "loss": 0.3172,
+      "epoch": 0.3199658703071672,
+      "grad_norm": 0.0022936267778277397,
+      "learning_rate": 4.600042662116041e-08,
+      "loss": 0.303,
       "step": 3000
     },
     {
-      "epoch": 0.552689756816507,
-      "eval_loss": 1.1506118774414062,
-      "eval_runtime": 74.8655,
-      "eval_samples_per_second": 16.122,
-      "eval_steps_per_second": 2.017,
+      "epoch": 0.3199658703071672,
+      "eval_loss": 1.2202869653701782,
+      "eval_runtime": 63.5857,
+      "eval_samples_per_second": 15.522,
+      "eval_steps_per_second": 1.95,
       "step": 3000
     },
     {
-      "epoch": 0.6448047162859248,
-      "grad_norm": 65.59048461914062,
-      "learning_rate": 4.193994104642594e-08,
-      "loss": 0.2751,
+      "epoch": 0.37329351535836175,
+      "grad_norm": 15.805087089538574,
+      "learning_rate": 4.5333831058020476e-08,
+      "loss": 0.3242,
       "step": 3500
     },
     {
-      "epoch": 0.6448047162859248,
-      "eval_loss": 1.1169357299804688,
-      "eval_runtime": 76.2936,
-      "eval_samples_per_second": 15.82,
-      "eval_steps_per_second": 1.979,
+      "epoch": 0.37329351535836175,
+      "eval_loss": 1.195627212524414,
+      "eval_runtime": 63.4024,
+      "eval_samples_per_second": 15.567,
+      "eval_steps_per_second": 1.956,
       "step": 3500
     },
     {
-      "epoch": 0.7369196757553427,
-      "grad_norm": 18.106201171875,
-      "learning_rate": 4.078850405305821e-08,
-      "loss": 0.2919,
+      "epoch": 0.42662116040955633,
+      "grad_norm": 5.377908229827881,
+      "learning_rate": 4.4667235494880546e-08,
+      "loss": 0.2427,
       "step": 4000
     },
     {
-      "epoch": 0.7369196757553427,
-      "eval_loss": 1.0920671224594116,
-      "eval_runtime": 74.9101,
-      "eval_samples_per_second": 16.113,
-      "eval_steps_per_second": 2.016,
+      "epoch": 0.42662116040955633,
+      "eval_loss": 1.169384241104126,
+      "eval_runtime": 63.5082,
+      "eval_samples_per_second": 15.541,
+      "eval_steps_per_second": 1.953,
       "step": 4000
     },
     {
-      "epoch": 0.8290346352247605,
-      "grad_norm": 506.40570068359375,
-      "learning_rate": 3.9637067059690496e-08,
-      "loss": 0.2583,
+      "epoch": 0.47994880546075086,
+      "grad_norm": 150.5334930419922,
+      "learning_rate": 4.4000639931740615e-08,
+      "loss": 0.2993,
       "step": 4500
     },
     {
-      "epoch": 0.8290346352247605,
-      "eval_loss": 1.0721209049224854,
-      "eval_runtime": 75.8053,
-      "eval_samples_per_second": 15.922,
-      "eval_steps_per_second": 1.992,
+      "epoch": 0.47994880546075086,
+      "eval_loss": 1.145558476448059,
+      "eval_runtime": 62.3888,
+      "eval_samples_per_second": 15.82,
+      "eval_steps_per_second": 1.988,
       "step": 4500
     },
     {
-      "epoch": 0.9211495946941783,
-      "grad_norm": 1.1508910655975342,
-      "learning_rate": 3.848563006632277e-08,
-      "loss": 0.2679,
+      "epoch": 0.5332764505119454,
+      "grad_norm": 0.12246542423963547,
+      "learning_rate": 4.333404436860068e-08,
+      "loss": 0.3183,
       "step": 5000
     },
     {
-      "epoch": 0.9211495946941783,
-      "eval_loss": 1.0519349575042725,
-      "eval_runtime": 75.226,
-      "eval_samples_per_second": 16.045,
-      "eval_steps_per_second": 2.007,
+      "epoch": 0.5332764505119454,
+      "eval_loss": 1.1201218366622925,
+      "eval_runtime": 63.7154,
+      "eval_samples_per_second": 15.491,
+      "eval_steps_per_second": 1.946,
       "step": 5000
     },
     {
-      "epoch": 1.013264554163596,
-      "grad_norm": 0.1393290013074875,
-      "learning_rate": 3.733419307295505e-08,
-      "loss": 0.2472,
+      "epoch": 0.58660409556314,
+      "grad_norm": 0.3306196630001068,
+      "learning_rate": 4.266744880546075e-08,
+      "loss": 0.307,
       "step": 5500
     },
     {
-      "epoch": 1.013264554163596,
-      "eval_loss": 1.0355629920959473,
-      "eval_runtime": 74.6535,
-      "eval_samples_per_second": 16.168,
-      "eval_steps_per_second": 2.023,
+      "epoch": 0.58660409556314,
+      "eval_loss": 1.0982334613800049,
+      "eval_runtime": 62.4882,
+      "eval_samples_per_second": 15.795,
+      "eval_steps_per_second": 1.984,
       "step": 5500
     },
     {
-      "epoch": 1.105379513633014,
-      "grad_norm": 200.34664916992188,
-      "learning_rate": 3.6182756079587324e-08,
-      "loss": 0.26,
+      "epoch": 0.6399317406143344,
+      "grad_norm": 149.38189697265625,
+      "learning_rate": 4.2000853242320816e-08,
+      "loss": 0.2638,
       "step": 6000
     },
     {
-      "epoch": 1.105379513633014,
-      "eval_loss": 1.0176944732666016,
-      "eval_runtime": 74.4374,
-      "eval_samples_per_second": 16.215,
-      "eval_steps_per_second": 2.029,
+      "epoch": 0.6399317406143344,
+      "eval_loss": 1.0779541730880737,
+      "eval_runtime": 63.4101,
+      "eval_samples_per_second": 15.565,
+      "eval_steps_per_second": 1.956,
       "step": 6000
     },
     {
-      "epoch": 1.1974944731024317,
-      "grad_norm": 431.1804504394531,
-      "learning_rate": 3.50313190862196e-08,
-      "loss": 0.2153,
+      "epoch": 0.693259385665529,
+      "grad_norm": 353.0036315917969,
+      "learning_rate": 4.1334257679180886e-08,
+      "loss": 0.2226,
       "step": 6500
     },
     {
-      "epoch": 1.1974944731024317,
-      "eval_loss": 1.0044586658477783,
-      "eval_runtime": 74.3247,
-      "eval_samples_per_second": 16.24,
-      "eval_steps_per_second": 2.032,
+      "epoch": 0.693259385665529,
+      "eval_loss": 1.0613036155700684,
+      "eval_runtime": 62.8292,
+      "eval_samples_per_second": 15.709,
+      "eval_steps_per_second": 1.974,
       "step": 6500
     },
     {
-      "epoch": 1.2896094325718497,
-      "grad_norm": 0.6204941272735596,
-      "learning_rate": 3.387988209285188e-08,
-      "loss": 0.1791,
+      "epoch": 0.7465870307167235,
+      "grad_norm": 0.13188917934894562,
+      "learning_rate": 4.0667662116040955e-08,
+      "loss": 0.2453,
       "step": 7000
     },
     {
-      "epoch": 1.2896094325718497,
-      "eval_loss": 0.9926707148551941,
-      "eval_runtime": 75.7317,
-      "eval_samples_per_second": 15.938,
-      "eval_steps_per_second": 1.994,
+      "epoch": 0.7465870307167235,
+      "eval_loss": 1.0443507432937622,
+      "eval_runtime": 62.7457,
+      "eval_samples_per_second": 15.73,
+      "eval_steps_per_second": 1.976,
       "step": 7000
     },
     {
-      "epoch": 1.3817243920412676,
-      "grad_norm": 52.27273178100586,
-      "learning_rate": 3.272844509948415e-08,
-      "loss": 0.2082,
+      "epoch": 0.7999146757679181,
+      "grad_norm": 59.391639709472656,
+      "learning_rate": 4.000106655290102e-08,
+      "loss": 0.272,
       "step": 7500
     },
     {
-      "epoch": 1.3817243920412676,
-      "eval_loss": 0.9803809523582458,
-      "eval_runtime": 74.9936,
-      "eval_samples_per_second": 16.095,
-      "eval_steps_per_second": 2.014,
+      "epoch": 0.7999146757679181,
+      "eval_loss": 1.0300519466400146,
+      "eval_runtime": 62.3719,
+      "eval_samples_per_second": 15.824,
+      "eval_steps_per_second": 1.988,
       "step": 7500
     },
     {
-      "epoch": 1.4738393515106853,
-      "grad_norm": 0.0976635068655014,
-      "learning_rate": 3.1577008106116435e-08,
-      "loss": 0.196,
+      "epoch": 0.8532423208191127,
+      "grad_norm": 495.4797668457031,
+      "learning_rate": 3.9334470989761093e-08,
+      "loss": 0.283,
       "step": 8000
     },
     {
-      "epoch": 1.4738393515106853,
-      "eval_loss": 0.9711655378341675,
-      "eval_runtime": 74.3537,
-      "eval_samples_per_second": 16.233,
-      "eval_steps_per_second": 2.031,
+      "epoch": 0.8532423208191127,
+      "eval_loss": 1.0166898965835571,
+      "eval_runtime": 62.0338,
+      "eval_samples_per_second": 15.911,
+      "eval_steps_per_second": 1.999,
       "step": 8000
     },
     {
-      "epoch": 1.565954310980103,
-      "grad_norm": 46.64162826538086,
-      "learning_rate": 3.042557111274871e-08,
-      "loss": 0.1946,
+      "epoch": 0.9065699658703071,
+      "grad_norm": 159.0310821533203,
+      "learning_rate": 3.866787542662116e-08,
+      "loss": 0.2331,
       "step": 8500
     },
     {
-      "epoch": 1.565954310980103,
-      "eval_loss": 0.9621181488037109,
-      "eval_runtime": 75.7205,
-      "eval_samples_per_second": 15.94,
-      "eval_steps_per_second": 1.994,
+      "epoch": 0.9065699658703071,
+      "eval_loss": 1.003544807434082,
+      "eval_runtime": 62.9704,
+      "eval_samples_per_second": 15.674,
+      "eval_steps_per_second": 1.969,
       "step": 8500
     },
     {
-      "epoch": 1.658069270449521,
-      "grad_norm": 0.05690698325634003,
-      "learning_rate": 2.927413411938099e-08,
-      "loss": 0.2422,
+      "epoch": 0.9598976109215017,
+      "grad_norm": 45.40327835083008,
+      "learning_rate": 3.8001279863481225e-08,
+      "loss": 0.2362,
       "step": 9000
     },
     {
-      "epoch": 1.658069270449521,
-      "eval_loss": 0.9536784291267395,
-      "eval_runtime": 75.1745,
-      "eval_samples_per_second": 16.056,
-      "eval_steps_per_second": 2.009,
+      "epoch": 0.9598976109215017,
+      "eval_loss": 0.992453396320343,
+      "eval_runtime": 63.1405,
+      "eval_samples_per_second": 15.632,
+      "eval_steps_per_second": 1.964,
       "step": 9000
     },
     {
-      "epoch": 1.750184229918939,
-      "grad_norm": 0.08470064401626587,
-      "learning_rate": 2.8122697126013263e-08,
-      "loss": 0.2106,
+      "epoch": 1.0132252559726962,
+      "grad_norm": 621.8538818359375,
+      "learning_rate": 3.7334684300341295e-08,
+      "loss": 0.2396,
       "step": 9500
     },
     {
-      "epoch": 1.750184229918939,
-      "eval_loss": 0.9457550644874573,
-      "eval_runtime": 74.5062,
-      "eval_samples_per_second": 16.2,
-      "eval_steps_per_second": 2.027,
+      "epoch": 1.0132252559726962,
+      "eval_loss": 0.9829725027084351,
+      "eval_runtime": 62.3244,
+      "eval_samples_per_second": 15.836,
+      "eval_steps_per_second": 1.99,
       "step": 9500
     },
     {
-      "epoch": 1.8422991893883567,
-      "grad_norm": 1.5745967626571655,
-      "learning_rate": 2.697126013264554e-08,
-      "loss": 0.1801,
+      "epoch": 1.0665529010238908,
+      "grad_norm": 461.60736083984375,
+      "learning_rate": 3.6668088737201364e-08,
+      "loss": 0.2013,
       "step": 10000
     },
     {
-      "epoch": 1.8422991893883567,
-      "eval_loss": 0.9393072724342346,
-      "eval_runtime": 75.9432,
-      "eval_samples_per_second": 15.893,
-      "eval_steps_per_second": 1.988,
+      "epoch": 1.0665529010238908,
+      "eval_loss": 0.9735796451568604,
+      "eval_runtime": 63.0763,
+      "eval_samples_per_second": 15.648,
+      "eval_steps_per_second": 1.966,
       "step": 10000
     }
   ],
   "logging_steps": 500,
-  "max_steps": 21712,
+  "max_steps": 37504,
   "num_input_tokens_seen": 0,
   "num_train_epochs": 4,
   "save_steps": 500,

checkpoint-10000/training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9a29c2374e59ce3f57b34db9cb20d2d1bdc870c3dad128e9a4d462636c37e8b0
+oid sha256:87e35dca972e13467c750106c19301b098fbd2918fda1087487e227bac3aca37
 size 4847