changing loading utils for tf compatibilit

kundajelab · Apr 11, 2023 · d501bf9 · d501bf9
1 parent 93aaf2d
commit d501bf9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chrombpnet/evaluation/interpret/input_utils.py b/chrombpnet/evaluation/interpret/input_utils.py
@@ -31,7 +31,7 @@ def load_model_wrapper(args):
     # read .h5 model
     custom_objects={"multinomial_nll": losses.multinomial_nll, "tf": tf}    
     get_custom_objects().update(custom_objects)    
-    model=load_model(args.model_h5)
+    model=load_model(args.model_h5,compile=False)
     print("got the model")
     model.summary()
     return model