search

FMInference / DejaVu

268 stars 32 forks source link

fix file name format as they are named like *_sp_x_* #5

Open bilgeacun opened 11 months ago

bilgeacun commented 11 months ago

Minor issue in file names of the data dumps:

diff --git a/sparse_predictor/main_att.py b/sparse_predictor/main_att.py
index 909fce2..ebe87ac 100644
--- a/sparse_predictor/main_att.py
+++ b/sparse_predictor/main_att.py
@@ -69,7 +69,7 @@ class BasicDataset(Dataset):

 def get_data(args, l):
     if CONFIG[args.model]['ckt_storage'] == "bylayer":
-        path = f"{DATA[args.model][args.dataset]}/att_x_{l-1}.mmap"
+        path = f"{DATA[args.model][args.dataset]}/att_sp_x_{l-1}.mmap"