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

Implement KerasCallback for Attentions #22

Open keisen opened 4 years ago

keisen commented 4 years ago

https://www.tensorflow.org/guide/keras/custom_callback