make keras-vis compatible with TF2.1

raghakot / keras-vis

Neural network visualization toolkit for keras

https://raghakot.github.io/keras-vis

MIT License

2.98k stars 660 forks source link

make keras-vis compatible with TF2.1 #220

Open macfly1202 opened 4 years ago

macfly1202 commented 4 years ago

I make minor change to use TF1 behavior with Keras-Vis. Some warning still exists but no blocking

keisen commented 4 years ago

Thank you for this PR!

Before we merge this PR, could you improve something that is to keep comatibility with Keras that is original, not tf.keras. As following page, Keras support TF1. and TF2.0., so we believe that keras-vis should also support TF1.* .

https://github.com/keras-team/keras#multi-backend-keras-and-tfkeras

makquel commented 3 years ago

This minor change helps to deal compatibility error from the most recent keras wrapper API.

AttributeError: module 'tensorflow' has no attribute 'get_default_graph'

However, other errors appear with TensorFlow back end.