update list of loaded layers after pruning
This commit is contained in:
parent
fd1de882ba
commit
ddf1d54304
@ -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
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user