After an export of pytorch pretrained resnet101, and it conversion into savedModel with onnx-tf, it seems that even if SignatureDefs are present, ConcreteFunction are not well converted :
Concrete Functions:
Function Name: '__call__'
Named Argument #1
input
Function Name: 'gen_tensor_dict'
And then, if i want to save the model by redefining the SignatureDefs, the saving never stop and seems frozen :
Describe the bug
After an export of pytorch pretrained resnet101, and it conversion into savedModel with onnx-tf, it seems that even if SignatureDefs are present, ConcreteFunction are not well converted :
And then, if i want to save the model by redefining the SignatureDefs, the saving never stop and seems frozen :
t_spec = tf.TensorSpec(shape=(None, 3, 32, 32), dtype=tf.float32, name='input') c_func = loaded_model.__call__.get_concrete_function(input=t_spec)
To Reproduce
Please give us instructions to reproduce your problem.
Python, ONNX, ONNX-TF, Tensorflow version
This section can be obtained by running
get_version.py
from util folder.Additional context
Even if i try to reimplement the call method, i am not able to call model(input_tensor).