From 8a3d7647d23aae9a41df0d41d5f9990bae116557 Mon Sep 17 00:00:00 2001
From: Luo Tao <luotao02@baidu.com>
Date: Thu, 15 Mar 2018 21:43:33 +0800
Subject: [PATCH] fix AttributeError: 'module' object has no attribute
 'framework_pb2'

---
 python/paddle/fluid/layers/layer_function_generator.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/python/paddle/fluid/layers/layer_function_generator.py b/python/paddle/fluid/layers/layer_function_generator.py
index bd79022a0c3..35b01a79914 100644
--- a/python/paddle/fluid/layers/layer_function_generator.py
+++ b/python/paddle/fluid/layers/layer_function_generator.py
@@ -16,10 +16,7 @@ import cStringIO
 import functools
 import warnings
 
-from .. import proto
-
-framework_pb2 = proto.framework_pb2
-
+from ..proto import framework_pb2
 from ..framework import OpProtoHolder, Variable
 from ..layer_helper import LayerHelper
 
-- 
GitLab