fix model printing error

bcgsc · Jul 3, 2022 · 1ada763 · 1ada763
1 parent cae4611
commit 1ada763
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/AMPlify.py b/src/AMPlify.py
@@ -176,10 +176,12 @@ def main():
 
     args = parser.parse_args()
 
-    print('\nLoading \s models...'%args.model)
     model_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))+'/models/'
     models = [model_dir + args.model + '/AMPlify_' + args.model + '_model_weights_' \
               + str(i+1) + '.h5' for i in range(5)]
+    print('\nLoading %s models...'%args.model)
+    for n in range(5):
+        print(models[n])
     # load models for final output
     out_model = load_multi_model(models, build_amplify)
     # load_models for attention