Allan Victor
commited on
Commit
·
d8795c2
1
Parent(s):
fac590f
Update Util_funs.py
Browse files- Util_funs.py +1 -1
Util_funs.py
CHANGED
@@ -242,7 +242,7 @@ def train_loop(data_train_loader, data_test_loader, model, device, epoch = 4, lr
|
|
242 |
input_ids, attention_mask,q_token_type_ids, label_id = batch
|
243 |
|
244 |
# Predictions
|
245 |
-
|
246 |
|
247 |
# prediction = prediction.detach().cpu().squeeze()
|
248 |
# label_id = label_id.detach().cpu()
|
|
|
242 |
input_ids, attention_mask,q_token_type_ids, label_id = batch
|
243 |
|
244 |
# Predictions
|
245 |
+
_, feature, _ = model_meta(input_ids, attention_mask,q_token_type_ids, labels = label_id.squeeze())
|
246 |
|
247 |
# prediction = prediction.detach().cpu().squeeze()
|
248 |
# label_id = label_id.detach().cpu()
|