Add pruning walker. (PaddlePaddle#5)

lizexu123 · Jan 3, 2020 · 790a9ff · 790a9ff
1 parent cc1cd6f
commit 790a9ff
Show file tree

Hide file tree

Showing 11 changed files with 1,226 additions and 568 deletions.
diff --git a/demo/models/__init__.py b/demo/models/__init__.py
@@ -1,5 +1,6 @@
 from .mobilenet import MobileNet
 from .resnet import ResNet34, ResNet50
 from .mobilenet_v2 import MobileNetV2
+from .pvanet import PVANet
 
-__all__ = ['MobileNet', 'ResNet34', 'ResNet50', 'MobileNetV2']
+__all__ = ['MobileNet', 'ResNet34', 'ResNet50', 'MobileNetV2', 'PVANet']