Merge pull request #9128 from luotao1/layer_function_generator

fix AttributeError: 'module' object has no attribute 'framework_pb2'
PaddlePaddle · Mar 16, 2018 · 4f3b73e · 4f3b73e · QiaoShin · Jul 18, 2022
2 parents a448fbe + 8a3d764
commit 4f3b73e
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/paddle/fluid/layers/layer_function_generator.py b/python/paddle/fluid/layers/layer_function_generator.py
@@ -16,10 +16,7 @@
 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