fix bert label dataype

undertherain · Jun 19, 2020 · 9fba12c · 9fba12c
1 parent f699ba5
commit 9fba12c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarker/modules/problems/bert/data.py b/benchmarker/modules/problems/bert/data.py
@@ -10,5 +10,5 @@ def get_data(params):
              params["problem"]["len_sequence"],
              )
     X = np.random.random(shape).astype(np.int64)
-    Y = np.ones((cnt_batches, params["batch_size"]))
+    Y = np.ones((cnt_batches, params["batch_size"]), dtype=np.int64)
     return X, Y