search

keisen / tf-keras-vis

Neural network visualization toolkit for tf.keras
https://keisen.github.io/tf-keras-vis-docs/
MIT License
313 stars 45 forks source link

Update model_modifiers.py #91

Closed Lakshay-13 closed 1 year ago

Lakshay-13 commented 2 years ago

To address the error -

ImportError: cannot import name 'Conv' from 'keras.layers.convolutional' (.../lib/python3.10/site-packages/keras/layers/convolutional/init.py) (also mentioned at https://github.com/keisen/tf-keras-vis/issues/87)

This fixes the if statement allowing TensorFlow versions newer than "2.6.0rc0" to import from tensorflow.python.keras.layers.convolutional instead of keras.layers.convolutional