Update synth data pipeline dtype

ffbada72 · guptapriya · 13cc0f70 · ffbada72
Commit ffbada72 authored Jul 18, 2019 by guptapriya
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/transformer/v2/data_pipeline.py official/transformer/v2/data_pipeline.py +2 -2

No files found.
--- a/official/transformer/v2/data_pipeline.py
+++ b/official/transformer/v2/data_pipeline.py
@@ -268,10 +268,10 @@ def _generate_synthetic_data(params):
  dataset = model_helpers.generate_synthetic_data(
      input_shape=tf.TensorShape([length]),
      input_value=1,
-      input_dtype=tf.int32,
+      input_dtype=tf.int64,
      label_shape=tf.TensorShape([length]),
      label_value=1,
-      label_dtype=tf.int32,
+      label_dtype=tf.int64,
  )
  return dataset.batch(batch)