提交 b220a63e 编写于 作者: W wjj19950828

fixed import models bug

上级 3c94431c
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
import paddle import paddle
from paddle.fluid.core import VarBase from paddle.framework import core
from x2paddle.utils import paddle_dtypes from x2paddle.utils import paddle_dtypes
...@@ -94,9 +94,9 @@ def __getitem__(self, idx): ...@@ -94,9 +94,9 @@ def __getitem__(self, idx):
else: else:
return out return out
VarBase = core.eager.Tensor
VarBase.__getitem__ = __getitem__ VarBase.__getitem__ = __getitem__
VarBase.setitem_tmp = VarBase.__setitem__ VarBase.setitem_tmp = VarBase.__setitem__
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册