Fix bug

pomonam · Mar 20, 2024 · 8b3405b · 8b3405b
1 parent c38da1f
commit 8b3405b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/glue/README.md b/examples/glue/README.md
@@ -24,9 +24,9 @@ python train.py --dataset_name sst2 \
 
 To obtain a pairwise influence scores on maximum of 2000 query data points using `ekfac`, run the following command:
 ```bash
-python analyze.py --query_batch_size 8 \
+python analyze.py --dataset_name sst2 \
+    --query_batch_size 8 \
     --train_batch_size 32 \
-    --dataset_dir ./data \
     --checkpoint_dir ./checkpoints \
     --factor_strategy ekfac
 ```

diff --git a/examples/glue/analyze.py b/examples/glue/analyze.py
@@ -117,11 +117,11 @@ def main():
     logging.basicConfig(level=logging.INFO)
 
     train_dataset = get_glue_dataset(
-        data_name=args.data_name,
+        data_name=args.dataset_name,
         split="eval_train",
     )
     eval_dataset = get_glue_dataset(
-        data_name=args.data_name,
+        data_name=args.dataset_name,
         split="valid",
     )