Support for TF2 and tf.keras?

haidark / WaveletDeconv

Neural network layer code to implement wavelet deconvolutions

34 stars 11 forks source link

Support for TF2 and tf.keras? #2

Closed vlawhern closed 4 years ago

vlawhern commented 5 years ago

Hi,

I came across this package and found it pretty interesting. I was wondering if you were planning support for TF2, specifically with tf.keras. I was able to get this to work with regular Keras but was unable to with tf.keras.

haidark commented 5 years ago

Hi, thanks for finding this! Sorry, no plans to build support for TF2 and tf.keras but you are welcome to make a pull request :)