Browse Source

!4591 fixed the THOR optimizer train failed

Merge pull request !4591 from wangmin0104/master
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d884ae2da9
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      model_zoo/official/cv/resnet_thor/src/model_thor.py

+ 1
- 2
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



Loading…
Cancel
Save