other changes

delphi-suite · Feb 13, 2024 · d36a135 · d36a135
1 parent b0042b5
commit d36a135
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 185 deletions.
diff --git a/.gitignore b/.gitignore
@@ -159,4 +159,4 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
-Correct_logprobs1/
+Correct_logprobs/
diff --git a/scripts/inference.py b/scripts/inference.py
@@ -43,7 +43,6 @@ def main(model_name, dataset_split, batch_size):
 
     # model accepts 2D tensors (batch_size, seq_len)
     val_sequences = torch.tensor([s["tokens"] for s in val_ds])
-    val_sequences = val_sequences[:220]
 
     output_folder = "Correct_logprobs"
     os.makedirs(output_folder, exist_ok=True)

diff --git a/src/delphi/eval/generate_logprobs.sh b/src/delphi/eval/generate_logprobs.sh
diff --git a/src/delphi/eval/inference.py b/src/delphi/eval/inference.py
diff --git a/src/delphi/eval/inference_on_validation.py b/src/delphi/eval/inference_on_validation.py