Fix cuda compability for evaluation

2018-11-02 01:39:15 -04:00
parent 1d8511f8f2
commit 1d53f9cb72
1 changed files with 1 additions and 0 deletions
--- 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)