Why don't you use TensorFlow 2 for this code ? And have you been able to convert this code in tf2
Why don't you use TensorFlow 2 for this code ? And have you been able to convert this code in tf2