Training in progress, epoch 2, checkpoint

last-checkpoint/README.md

@@ -101,9 +101,9 @@ print(embeddings.shape)
 # Get the similarity scores for the embeddings
 similarities = model.similarity(embeddings, embeddings)
 print(similarities)
-# tensor([[1.0000, 0.7931, 0.5211],
-#         [0.7931, 1.0000, 0.5107],
-#         [0.5211, 0.5107, 1.0000]])
+# tensor([[1.0000, 0.7366, 0.4534],
+#         [0.7366, 1.0000, 0.4396],
+#         [0.4534, 0.4396, 1.0000]])
 ```
 
 <!--

last-checkpoint/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be363c432041c4a7573d3da2f9a0ab1c7df1a8b1a576928196c93327cad7f952
+oid sha256:de1414049a746b2e56574aabef3303b914e7b9a0f4c442d93140d5bdb4a130cc
 size 90864192

last-checkpoint/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a9302412d849e70b757484c545ee8f4200610e2b5f68eb68fb20afc6c651ca06
+oid sha256:168f24d2777ab0fe7d549b5cbc57061de102566dcc627477843b4c54d872bad6
 size 180607738

last-checkpoint/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:57f43045c242fd2557243ce9549745ee0255495a3fdd06cf3b3892bfec8fcd06
+oid sha256:d504d17d70cbf3694e88be868534a2e1dd8fdc9f2518c1e988f6b7f7aa42bff4
 size 14244

last-checkpoint/scaler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3c4c7b163f9ecd5aa5916a8a707d0281dceeb0bb3b46ace886ee05988e38d2f2
+oid sha256:5d473dbc02d3ee7bd9625ff87a50bb89b5cfc0fd4775012466ec6bee2c111c2e
 size 988

last-checkpoint/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0b1b14dcd842d795da6246093c7d462c99b0c308d1d89f50ec8d2d8c7a0bf69e
+oid sha256:c0b735d1c7f640d3ef4cf28a2af2021687802dd1b42e9507ad99c430c545922c
 size 1064

last-checkpoint/trainer_state.json

@@ -2,9 +2,9 @@
   "best_global_step": null,
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 1.0,
+  "epoch": 2.0,
   "eval_steps": 5000,
-  "global_step": 2278,
+  "global_step": 4556,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,