update list of loaded layers after pruning

This commit is contained in:
Jan Petykiewicz 2022-04-05 18:33:22 -07:00
parent fd1de882ba
commit ddf1d54304

View File

@ -155,6 +155,7 @@ def trace_connectivity(
finished_layers = loaded_layers - remaining_layers finished_layers = loaded_layers - remaining_layers
for layer in finished_layers: for layer in finished_layers:
nets_info.prune(layer) nets_info.prune(layer)
loaded_layers.remove(layer)
return nets_info return nets_info