提交 59ec0d64 编写于 作者: W wangmin

remove to_full_tensor for thor optimizer

上级 95fa3c52
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册