Merge pull request #9128 from luotao1/layer_function_generator

fix AttributeError: 'module' object has no attribute 'framework_pb2'

Merge pull request #9128 from luotao1/layer_function_generator
fix AttributeError: 'module' object has no attribute 'framework_pb2'
4f3b73ed · Tao Luo · GitHub · a448fbe9 · 8a3d7647 · 4f3b73ed
显示空白变更内容
内联并排

Showing with 1 addition and 4 deletion

python/paddle/fluid/layers/layer_function_generator.py python/paddle/fluid/layers/layer_function_generator.py +1 -4

未找到文件。
--- 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