From 1d53f9cb7244e242c3eee858948607a35ed5d3cc Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Fri, 2 Nov 2018 01:39:15 -0400 Subject: [PATCH] Fix cuda compability for evaluation --- run_classifier_pytorch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/run_classifier_pytorch.py b/run_classifier_pytorch.py index 64f8a74717..943b1be18c 100644 --- a/run_classifier_pytorch.py +++ b/run_classifier_pytorch.py @@ -552,6 +552,7 @@ def main(): input_ids = input_ids.to(device) input_mask = input_mask.float().to(device) segment_ids = segment_ids.to(device) + label_ids = label_ids.to(device) tmp_eval_loss, logits = model(input_ids, segment_ids, input_mask, label_ids)