Merge pull request #21 from jradrion/dnn_fix

fixed dnn error in predict

Author: jradrion

ReLERNN/helpers.py

@@ -265,6 +265,14 @@ def load_and_predictVCF(VCFGenerator,
 
     os.environ["CUDA_VISIBLE_DEVICES"] = str(gpuID)
 
+    ## The following code block appears necessary for running with tf2 and cudnn
+    from tensorflow.compat.v1 import ConfigProto
+    from tensorflow.compat.v1 import Session
+    config = ConfigProto()
+    config.gpu_options.allow_growth = True
+    Session(config=config)
+    ###
+
     # load json and create model
     if(network != None):
         jsonFILE = open(network[0],"r")