p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove to_full_tensor for thor optimizer

This commit is contained in:
wangmin 2020-08-17 17:22:58 +08:00
parent 95fa3c5202
commit 59ec0d6485
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/cv/resnet_thor/src/model_thor.py
View File

@ -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