fix quick start (PaddlePaddle#390)

lizexu123 · Jul 22, 2020 · 92b3e1c · 92b3e1c
1 parent 115bbda
commit 92b3e1c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/docs/en/quick_start/distillation_tutorial_en.md b/docs/en/quick_start/distillation_tutorial_en.md
@@ -25,7 +25,7 @@ This tutorial trains and verifies distillation model on the MNIST dataset. The i
 Select `ResNet50` as the teacher to perform distillation training on the students of the` MobileNet` architecture.
 
 ```python
-model = models.__dict__['MobileNet']()
+model = slim.models.MobileNet()
 student_program = fluid.Program()
 student_startup = fluid.Program()
 with fluid.program_guard(student_program, student_startup):
@@ -42,7 +42,7 @@ with fluid.program_guard(student_program, student_startup):
 
 
 ```python
-teacher_model = models.__dict__['ResNet50']()
+model = slim.models.ResNet50()
 teacher_program = fluid.Program()
 teacher_startup = fluid.Program()
 with fluid.program_guard(teacher_program, teacher_startup):

diff --git a/docs/zh_cn/quick_start/distillation_tutorial.md b/docs/zh_cn/quick_start/distillation_tutorial.md
@@ -27,7 +27,7 @@ import paddleslim as slim
 选择`ResNet50`作为teacher对`MobileNet`结构的student进行蒸馏训练。
 
 ```python
-model = models.__dict__['MobileNet']()
+model = slim.models.MobileNet()
 student_program = fluid.Program()
 student_startup = fluid.Program()
 with fluid.program_guard(student_program, student_startup):
@@ -44,7 +44,7 @@ with fluid.program_guard(student_program, student_startup):
 
 
 ```python
-teacher_model = models.__dict__['ResNet50']()
+model = slim.models.ResNet50()
 teacher_program = fluid.Program()
 teacher_startup = fluid.Program()
 with fluid.program_guard(teacher_program, teacher_startup):

diff --git a/paddleslim/models/__init__.py b/paddleslim/models/__init__.py
@@ -16,4 +16,7 @@
 from .util import image_classification
 from .slimfacenet import SlimFaceNet_A_x0_60, SlimFaceNet_B_x0_75, SlimFaceNet_C_x0_75
 from .slim_mobilenet import SlimMobileNet_v1, SlimMobileNet_v2, SlimMobileNet_v3, SlimMobileNet_v4, SlimMobileNet_v5
-__all__ = ["image_classification"]
+from .mobilenet import MobileNet
+from .resnet import ResNet50
+
+__all__ = ["image_classification", "MobileNet", "ResNet50"]