search

odlgroup / odl

Operator Discretization Library https://odlgroup.github.io/odl/
Mozilla Public License 2.0
374 stars 105 forks source link

Compatibility of 'as_tensorflow_layer' with tensorflow 2.0 #1598

Open jaweriaamjad opened 3 years ago

jaweriaamjad commented 3 years ago

Hi, I want to implement a deep learning model for CT reconstruction that will require computing the gradients of the sinogram w.r.t to the original images. Just wanted to check if 'as_tensorflow_layer' is compatible with tf 2.0?

thanks

adler-j commented 3 years ago

Hi! Sadly I've not tried it, and I'd expect it to not work very well since as_tensorflow_layer uses the old graph-structure. I would be very willing to assist in any attempt to update it though, since this feature is a reasonably often requested.