Fix cuda compability for evaluation

1d53f9cb · VictorSanh · 1d8511f8 · 1d53f9cb
Commit 1d53f9cb authored Nov 02, 2018 by VictorSanh
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

run_classifier_pytorch.py run_classifier_pytorch.py +1 -0

No files found.
--- 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)