From 99ed9c782f74bf0d6cbab318a54903ec75040e1f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=B0=A4=E5=AE=89=E5=8D=87?= Date: Fri, 1 Jul 2022 01:08:39 +0000 Subject: [PATCH] =?UTF-8?q?=E5=9B=9E=E9=80=80=20'Pull=20Request=20!943=20:?= =?UTF-8?q?=20[bugfix]=20Fix=20serializaton.'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- torch_npu/utils/serialization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_npu/utils/serialization.py b/torch_npu/utils/serialization.py index ab8e3fd6a2..df973bdaad 100644 --- a/torch_npu/utils/serialization.py +++ b/torch_npu/utils/serialization.py @@ -74,7 +74,7 @@ def to_cpu(data): elif isinstance(value, torch.Tensor): copy_data[key] = value.cpu() elif isinstance(value, nn.Module): - if torch_npu._C.is_npu(next(value.parameters())): + if torch_npu._C.is_npu(next(data.parameters())): setattr(value, "mark_npu", True) copy_data[key] = value.cpu() else: -- Gitee