requires re-write for TF 2.x

[mirekphd]

Under TF 2.3 it fails early, at the imports stage:
ModuleNotFoundError: No module named 'tensorflow.contrib'

[mzgubic]

Hey, feel free to open a PR, happy to review

[chandrasekarnarayana]

Hi,
I would like to know has any progress has been made on the code migration from TF1 to TF2.
If you have found a workaround please do let me know

one of the suggestions online has been to use a dense layer with linear activation instead of the linear layer from tf.contrib.layer, is this correct?