search

tensorflow / tensorrt

TensorFlow/TensorRT integration
Apache License 2.0
736 stars 226 forks source link

Error converting multilingual universal sentence encoder to TensorRT. Input 1 of node StatefulPartitionedCall was passed float from statefulpartitionedcall_args_1:0 incompatible with expected resource. #224

Open sebastinajay opened 4 years ago

sebastinajay commented 4 years ago

I'm getting below error while converting multilingual universal sentence encoder(https://tfhub.dev/google/universal-sentence-encoder-multilingual/3) to TensorRT using TrtGraphConverterV2

InvalidArgumentError: Input 1 of node StatefulPartitionedCall was passed float from statefulpartitionedcall_args_1:0 incompatible with expected resource.

Below is the Colab link : https://colab.research.google.com/drive/1g8jHUMdpMgAQPv_kG3YJhP12SmSdHW-H?usp=sharing

Python Version: 3.6.9 TensorFlow Version: 2.3.0

lminer commented 3 years ago

I'm getting a similar issue.