coh mech

ccl_win.py

@@ -128,8 +128,12 @@ class ccl_win(evaluate.Measurement):
         else:
             device = "cuda" if torch.cuda.is_available() else "cpu"
 
+        cur_dir = os.path.dirname(__file__)
         tokenizer = AutoTokenizer.from_pretrained("roberta-large")
-        model = AutoModelForSequenceClassification.from_pretrained(f"./{dataset}/", num_labels=2,cache_dir=MODEL_CACHE_DIR)
+
+        pdb.set_trace()
+
+        model = AutoModelForSequenceClassification.from_pretrained(os.path.join(cur_dir,dataset) , num_labels=2,cache_dir=MODEL_CACHE_DIR)
         model.to(device)
 
         pred_list,len_by_sample = self.preprocess_adjacent_window(preds)