diff --git a/python/paddle/fluid/contrib/__init__.py b/python/paddle/fluid/contrib/__init__.py
index 7442059ba07b2ed1d7164b9be60b8bbc92fec651..ca10db0a5450e0a38159fe2e38b2926f6b1900a7 100644
--- a/python/paddle/fluid/contrib/__init__.py
+++ b/python/paddle/fluid/contrib/__init__.py
@@ -32,6 +32,8 @@ from . import utils
 from .utils import *
 from . import extend_optimizer
 from .extend_optimizer import *
+from . import model_stat
+from .model_stat import *
 
 __all__ = []
 __all__ += decoder.__all__