From d9d7d1a4625c95798b03bb0ef044dacaa15b968c Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Sun, 4 Nov 2018 21:25:36 +0100
Subject: [PATCH] update float()

---
 Comparing TF and PT models.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Comparing TF and PT models.ipynb b/Comparing TF and PT models.ipynb
index 18b18a80a4..a7be90f7e4 100644
--- a/Comparing TF and PT models.ipynb	
+++ b/Comparing TF and PT models.ipynb	
@@ -1015,7 +1015,7 @@
     "    print(input_mask)\n",
     "    print(example_indices)\n",
     "    input_ids = input_ids.to(device)\n",
-    "    input_mask = input_mask.float().to(device)\n",
+    "    input_mask = input_mask.to(device)\n",
     "\n",
     "    all_encoder_layers, _ = model(input_ids, token_type_ids=input_type_ids, attention_mask=input_mask)\n",
     "\n",