fixed import models bug

42995b86 · wjj19950828 · bae722ec · 42995b86
显示空白变更内容
内联并排

Showing with 3 addition and 2 deletion

x2paddle/project_convertor/pytorch/torch2paddle/varbase.py x2paddle/project_convertor/pytorch/torch2paddle/varbase.py +3 -2

未找到文件。
--- a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py
+++ b/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py
@@ -53,6 +53,8 @@ def is_condition_two(idx):
    return False


+VarBase = core.eager.Tensor
+
 VarBase.tmp = VarBase.__getitem__


@@ -95,9 +97,8 @@ def __getitem__(self, idx):
            return out


-VarBase = core.eager.Tensor
-
 VarBase.__getitem__ = __getitem__
+
 VarBase.setitem_tmp = VarBase.__setitem__