TensorFlow 2.x support?

gvtulder / elasticdeform

Differentiable elastic deformations for N-dimensional images (Python, SciPy, NumPy, TensorFlow, PyTorch).

Other

186 stars 25 forks source link

TensorFlow 2.x support? #8

Closed stnava closed 3 years ago

stnava commented 3 years ago

just curious if you are considering updating TF support to 2.x? may be fairly simple e.g. tensorflow.py_func ... tensorflow.py_function and removing stateful...

thanks!

gvtulder commented 3 years ago

Thanks for the suggestion. This should fix it I hope. I don't use TensorFlow myself anymore, but the tests pass.