Upload results

evaluate.py

@@ -8,7 +8,7 @@ from huggingface_hub import list_repo_refs
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 device = "cuda"
-test_indices_length = 10
+test_indices_length = 1000
 
 models = ["distributed/optimized-gpt2-250m", "distributed/optimized-gpt2-250m-v0.1.1", "distributed/gpt2-94m"]
 
@@ -28,7 +28,7 @@ for model_name in models:
     refs = list_repo_refs(model_name, repo_type="model")
     global_epoch = max([int(tag.name) for tag in refs.tags]) if refs.tags else None
 
-    for epoch in range(0,global_epoch):
+    for epoch in range(0,global_epoch, 5):
 
         if str(epoch) in results[model_name]['main-net'].keys():
             continue

results.json

@@ -771,6 +771,51 @@
             ],
             "1280": [
                 5.533918690816667
+            ],
+            "1285": [
+                5.567983552905323
+            ],
+            "1290": [
+                5.5455932653448885
+            ],
+            "1295": [
+                5.618938806755789
+            ],
+            "1300": [
+                5.57726935165956
+            ],
+            "1305": [
+                5.504214384279199
+            ],
+            "1310": [
+                5.591931197914315
+            ],
+            "1315": [
+                5.649649393850551
+            ],
+            "1320": [
+                5.618240403285898
+            ],
+            "1325": [
+                5.617241533815735
+            ],
+            "1330": [
+                5.576359107468154
+            ],
+            "1335": [
+                5.539896753655762
+            ],
+            "1340": [
+                5.518490235810309
+            ],
+            "1345": [
+                5.588169578686511
+            ],
+            "1350": [
+                5.50622625116791
+            ],
+            "1355": [
+                5.502286915139202
             ]
         },
         "baseline": {