Fix model_stat importing in paddle/fluid/contrib

6 years ago · fb1ae722d1
parent 21622ca30b
commit fb1ae722d1
1 changed files with 2 additions and 0 deletions
--- 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__