jan/snarled
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

update list of loaded layers after pruning

Browse source
This commit is contained in:
Jan Petykiewicz 2022-04-05 18:33:22 -07:00
commit ddf1d54304
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
snarled/main.py
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