create_keras_neuropod() does not work with TF 2.6 (AttributeError: 'KerasTensor' object has no attribute 'graph') - Githubissues

uber / neuropod

A uniform interface to run deep learning models from multiple frameworks

https://neuropod.ai

Apache License 2.0

936 stars 77 forks source link

create_keras_neuropod() does not work with TF 2.6 (AttributeError: 'KerasTensor' object has no attribute 'graph') #514

Closed thuningxu closed 3 years ago

thuningxu commented 3 years ago

Bug

  File "...neuropod/backends/keras/packager.py", line 85, in create_keras_neuropod
    graph_def = model.outputs[0].graph.as_graph_def()
AttributeError: 'KerasTensor' object has no attribute 'graph'

To Reproduce

Steps to reproduce the behavior:

This is test script running with Neuropod 0.3.0rc3 and TF 2.6 https://gist.github.com/thuningxu/659c8d4ed355e5c544a2bb2eb002415b

Expected behavior

Neuropod created and inference made

Environment

Neuropod Version (e.g., 0.2.0): 0.3.0rc3
OS (Linux, macOS): Linux
Language (Python, C++, Go bindings): Python
Python version: 3.6.9
Using OPE: no

If this bug report is about running a specific model:

Neuropod backend (e.g. TensorFlow, PyTorch, Keras, TorchScript, Python): Tensorflow
Framework version (e.g. 1.3.0): 2.6

If running on GPU:

CUDA/cuDNN version: 11.4
The output of nvidia-smi: NVIDIA-SMI 470.63.01 Driver Version: 470.63.01 CUDA Version: 11.4
Any other relevant information:

Additional context

thuningxu commented 3 years ago

I figured out this...basically I need to call tensorflow.disable_v2_behavior() before creating the Keras model. This way the model will not have KerasTensor.