Upload 4 files
Browse files- config.json +8 -0
- embedding_info.json +5 -0
- jepa_model.pt +3 -0
- training_log.txt +10 -0
config.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"embedding_model": "all-MiniLM-L6-v2",
|
| 3 |
+
"batch_size": 8,
|
| 4 |
+
"epochs": 10,
|
| 5 |
+
"learning_rate": 0.001,
|
| 6 |
+
"predictor_hidden_dim": 512,
|
| 7 |
+
"device": "cpu"
|
| 8 |
+
}
|
embedding_info.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "all-MiniLM-L6-v2",
|
| 3 |
+
"embedding_dim": 384,
|
| 4 |
+
"num_samples": 3
|
| 5 |
+
}
|
jepa_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bcae2a3ad0d7bdfe527133ce5b44743254bb8eefb77fac2de7499b369a7b8699
|
| 3 |
+
size 1579003
|
training_log.txt
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Epoch 1/10 Loss: 0.0003
|
| 2 |
+
Epoch 2/10 Loss: 0.0003
|
| 3 |
+
Epoch 3/10 Loss: 0.0003
|
| 4 |
+
Epoch 4/10 Loss: 0.0003
|
| 5 |
+
Epoch 5/10 Loss: 0.0003
|
| 6 |
+
Epoch 6/10 Loss: 0.0003
|
| 7 |
+
Epoch 7/10 Loss: 0.0003
|
| 8 |
+
Epoch 8/10 Loss: 0.0003
|
| 9 |
+
Epoch 9/10 Loss: 0.0003
|
| 10 |
+
Epoch 10/10 Loss: 0.0003
|