one line script

ThrunGroup · May 22, 2024 · e237683 · e237683
1 parent 66e66d3
commit e237683
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/experiments/run_llm.py b/experiments/run_llm.py
@@ -37,8 +37,8 @@ def run_llm(dataset, model, delta=0.01, eps=0.3):
       save_to=save_path,
       model=model_name,
       dataset=dataset,
-      A=A,
-      X=X,
+      A=A[:2, :2],
+      X=X[:2, :2],
       multiplicative_error=eps,
       failure_probability=delta,
       noise_bound=None,