From b220a63ef416db3f840798b53d151cccca1581f0 Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Wed, 12 Apr 2023 11:54:59 +0800 Subject: [PATCH] fixed import models bug --- x2paddle/project_convertor/pytorch/torch2paddle/varbase.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py b/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py index c159676..166653e 100644 --- a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py +++ b/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py @@ -13,7 +13,7 @@ # limitations under the License. import paddle -from paddle.fluid.core import VarBase +from paddle.framework import core from x2paddle.utils import paddle_dtypes @@ -94,9 +94,9 @@ def __getitem__(self, idx): else: return out +VarBase = core.eager.Tensor VarBase.__getitem__ = __getitem__ - VarBase.setitem_tmp = VarBase.__setitem__ -- GitLab