Skip to content

Commit 7c3a5af

Browse files
committed
README
1 parent 4b9df61 commit 7c3a5af

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

README.md

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,15 +41,19 @@ Pruning:
4141
TBD
4242
---
4343

44-
- Change the pruning to be done in one pass. Currently each of the 512 filters are pruned sequentually. This is inefficient since allocating new layers, especially fully connected layers with lots of parameters, is slow.
45-
In principle this can be done in a single pass.
44+
- Change the pruning to be done in one pass. Currently each of the 512 filters are pruned sequentually.
45+
`
46+
for layer_index, filter_index in prune_targets:
47+
model = prune_vgg16_conv_layer(model, layer_index, filter_index)
48+
`
4649

47-
{% highlight python %}
4850

49-
for layer_index, filter_index in prune_targets:
50-
model = prune_vgg16_conv_layer(model, layer_index, filter_index)
51+
This is inefficient since allocating new layers, especially fully connected layers with lots of parameters, is slow.
52+
53+
In principle this can be done in a single pass.
54+
5155

52-
{% endhighlight %}
5356

5457
- Change prune_vgg16_conv_layer to support additional architectures.
55-
The most immediate one would be VGG with batch norm.
58+
The most immediate one would be VGG with batch norm.
59+

0 commit comments

Comments
 (0)