update list of loaded layers after pruning

2022-04-05 18:33:22 -07:00 · 2022-04-05 18:33:22 -07:00 · ddf1d54304
commit ddf1d54304
parent fd1de882ba
1 changed files with 1 additions and 0 deletions
--- a/snarled/main.py
+++ b/snarled/main.py
@ -155,6 +155,7 @@ def trace_connectivity(
        finished_layers = loaded_layers - remaining_layers
        for layer in finished_layers:
            nets_info.prune(layer)
+            loaded_layers.remove(layer)

    return nets_info