Push LDS results pickle files

pomonam · Jun 26, 2024 · bd19cf0 · bd19cf0
1 parent 2b736bb
commit bd19cf0
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 3 deletions.
diff --git a/examples/wikitext/evaluate_lds.py b/examples/wikitext/evaluate_lds.py
@@ -10,7 +10,7 @@
 from kronfluence.analyzer import Analyzer
 
 
-def evaluate_correlations(scores):
+def evaluate_correlations(scores: torch.Tensor) -> float:
     losses = torch.from_numpy(torch.load(open("files/losses.pt", "rb")))
     masks = torch.from_numpy(torch.load(open("files/masks.pt", "rb"))).float()
 

diff --git a/examples/wikitext/files/losses.pt b/examples/wikitext/files/losses.pt
diff --git a/examples/wikitext/files/masks.pt b/examples/wikitext/files/masks.pt
diff --git a/examples/wikitext/requirements.txt b/examples/wikitext/requirements.txt
@@ -1,5 +1,4 @@
 transformers
 datasets
 matplotlib
-tueplots
-gdown
+tueplots