Fix for issue #56

3dem · Nov 22, 2023 · cef6dec · cef6dec
1 parent d258f2f
commit cef6dec
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/model_angelo/apps/build_no_seq.py b/model_angelo/apps/build_no_seq.py
@@ -196,8 +196,11 @@ def main(parsed_args):
             gnn_infer_args.refine = False
 
             if i == total_gnn_rounds - 1:
-                gnn_infer_args.aggressive_pruning = True
-
+                if parsed_args.config_path is None:
+                    gnn_infer_args.aggressive_pruning = True
+            else:
+                gnn_infer_args.aggressive_pruning = False
+
             logger.info(
                 f"GNN model refinement round {i + 1} with args: {gnn_infer_args}"
             )