update float()

d9d7d1a4 · thomwolf · c6207d85 · d9d7d1a4
Commit d9d7d1a4 authored Nov 04, 2018 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

Comparing TF and PT models.ipynb Comparing TF and PT models.ipynb +1 -1

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