Spaces:
Sleeping
Sleeping
ronald
commited on
Commit
·
6a45dac
1
Parent(s):
092069b
coh mech
Browse files- ccl_win.py +5 -1
ccl_win.py
CHANGED
@@ -128,8 +128,12 @@ class ccl_win(evaluate.Measurement):
|
|
128 |
else:
|
129 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
130 |
|
|
|
131 |
tokenizer = AutoTokenizer.from_pretrained("roberta-large")
|
132 |
-
|
|
|
|
|
|
|
133 |
model.to(device)
|
134 |
|
135 |
pred_list,len_by_sample = self.preprocess_adjacent_window(preds)
|
|
|
128 |
else:
|
129 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
130 |
|
131 |
+
cur_dir = os.path.dirname(__file__)
|
132 |
tokenizer = AutoTokenizer.from_pretrained("roberta-large")
|
133 |
+
|
134 |
+
pdb.set_trace()
|
135 |
+
|
136 |
+
model = AutoModelForSequenceClassification.from_pretrained(os.path.join(cur_dir,dataset) , num_labels=2,cache_dir=MODEL_CACHE_DIR)
|
137 |
model.to(device)
|
138 |
|
139 |
pred_list,len_by_sample = self.preprocess_adjacent_window(preds)
|