search

Bisonai / mobilenetv3-tensorflow

Unofficial implementation of MobileNetV3 architecture described in paper Searching for MobileNetV3.
Apache License 2.0
225 stars 81 forks source link

AttributeError: 'PrefetchDataset' object has no attribute 'output_shapes' #16

Open joshisr10 opened 4 years ago

joshisr10 commented 4 years ago

Could you please advise on this error?

I tried to resolve it through "tf.compat.v1.", but unsure of this new output_shapes error ---->

line 34, in build_dataset dataset["channels"] = ds_train.output_shapes["image"][-1].value

AttributeError: 'PrefetchDataset' object has no attribute 'output_shapes'

~Sanket

martinkersner commented 4 years ago

Hello @joshisr10,

Thank you for reaching out! Have you tried to run the code with TensorFlow 1.13?

Cheers, Martin

ayeshamkzhh commented 3 years ago

Tensorflow 1.13 also generates the same error.

manuelmele commented 3 years ago

I got the same error. Is there a workaround for this? Thanks in advance