odlgroup / odl

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

Does 'odl.contrib.tensorflow.as_tensorflow_layer' work for tensorflow v2.? #1583

Closed Donghyeon-Lee closed 4 years ago

Donghyeon-Lee commented 4 years ago

Hi, I'm trying to develop a reconstruction network using 'RayTransform' as a layer in odl. I checked that 'odl.contrib.tensorflow.as_tensorflow_layer' is worked for tensorflow v1.x and even GPU computing is available. However, in tensorflow v1.x some functions existing in v2 and needed in my network are missing. It seems 'odl.contrib.tensorflow' was developed 2 years ago, but tensorflow v2. was just made an year ago. So I want to get the comments from anyone have tried this issue before I try.

Thanks for the answer in advance.