From 59ec0d6485c382c5f05c07099308198b97114124 Mon Sep 17 00:00:00 2001 From: wangmin Date: Mon, 17 Aug 2020 17:22:58 +0800 Subject: [PATCH] remove to_full_tensor for thor optimizer --- model_zoo/official/cv/resnet_thor/src/model_thor.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/model_zoo/official/cv/resnet_thor/src/model_thor.py b/model_zoo/official/cv/resnet_thor/src/model_thor.py index 4ba6a32e9f..4433dd38ac 100644 --- a/model_zoo/official/cv/resnet_thor/src/model_thor.py +++ b/model_zoo/official/cv/resnet_thor/src/model_thor.py @@ -19,9 +19,8 @@ from mindspore.train.callback import RunContext from mindspore import context from mindspore import nn from mindspore.train.parallel_utils import ParallelMode -from mindspore.train._utils import _to_full_tensor from mindspore.train.model import Model -from mindspore.parallel._utils import _need_to_full +from mindspore.parallel._utils import _need_to_full, _to_full_tensor from mindspore.common.dtype import pytype_to_dtype from mindspore._c_expression import init_exec_dataset