update float()

2018-11-04 21:25:36 +01:00
parent c6207d85b6
commit d9d7d1a462
1 changed files with 1 additions and 1 deletions
--- a/models.ipynb
+++ b/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",