From d454014bd033d0278eaf6cc35d456cddee9940c7 Mon Sep 17 00:00:00 2001 From: Eric_lai Date: Fri, 2 Dec 2022 09:05:01 +0800 Subject: [PATCH 1/2] Bug fix --- ms_adapter/pytorch/nn/modules/container.py | 3 ++- setup.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/ms_adapter/pytorch/nn/modules/container.py b/ms_adapter/pytorch/nn/modules/container.py index c9fa2bbd..fa0e7ff1 100644 --- a/ms_adapter/pytorch/nn/modules/container.py +++ b/ms_adapter/pytorch/nn/modules/container.py @@ -4,6 +4,7 @@ from abc import abstractmethod from collections import OrderedDict, abc as container_abcs from mindspore.nn.layer.container import _get_prefix_and_index, _valid_index, _valid_cell from .module import Module +from ms_adapter.pytorch.tensor import cast_to_adapter_tensor class Sequential(Module): @@ -189,7 +190,7 @@ class Sequential(Module): def forward(self, input): for cell in self.cell_list: input = cell(input) - return input + return cast_to_adapter_tensor(input) class _ModuleListBase: """ diff --git a/setup.py b/setup.py index 2dacdc4f..b60c3b3d 100644 --- a/setup.py +++ b/setup.py @@ -23,7 +23,7 @@ def req_file(filename, folder=''): return [x.strip() for x in content] setup( - name="MSAdapter", + name="ms_adapter", version='.'.join(map(str, VERSION[:3])) + ''.join(VERSION[3:]), author="Peng Cheng Lab, HUAWEI", author_email="pcl.openi@pcl.ac.cn", -- 2.34.1 From e77dca8499575c2f98dd589dbc7efcec26d7ddd1 Mon Sep 17 00:00:00 2001 From: Eric_lai Date: Fri, 2 Dec 2022 10:32:52 +0800 Subject: [PATCH 2/2] fix pylint --- ms_adapter/pytorch/nn/modules/container.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ms_adapter/pytorch/nn/modules/container.py b/ms_adapter/pytorch/nn/modules/container.py index fa0e7ff1..cb44b288 100644 --- a/ms_adapter/pytorch/nn/modules/container.py +++ b/ms_adapter/pytorch/nn/modules/container.py @@ -3,8 +3,8 @@ from abc import abstractmethod from collections import OrderedDict, abc as container_abcs from mindspore.nn.layer.container import _get_prefix_and_index, _valid_index, _valid_cell -from .module import Module from ms_adapter.pytorch.tensor import cast_to_adapter_tensor +from .module import Module class Sequential(Module): -- 2.34.1