Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "b65f9b0fd565017a14d7d9ec64d4e83db5ee4158"
Commit ffbada72 authored by guptapriya's avatar guptapriya
Browse files

Update synth data pipeline dtype

parent 13cc0f70
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/transformer/v2/data_pipeline.py
View file @ ffbada72
...@@ -268,10 +268,10 @@ def _generate_synthetic_data(params): ...@@ -268,10 +268,10 @@ def _generate_synthetic_data(params):
dataset = model_helpers.generate_synthetic_data( dataset = model_helpers.generate_synthetic_data(
input_shape=tf.TensorShape([length]), input_shape=tf.TensorShape([length]),
input_value=1, input_value=1,
input_dtype=tf.int32, input_dtype=tf.int64,
label_shape=tf.TensorShape([length]), label_shape=tf.TensorShape([length]),
label_value=1, label_value=1,
label_dtype=tf.int32, label_dtype=tf.int64,
) )
return dataset.batch(batch) return dataset.batch(batch)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment