Browse Source

!6036 Repair loss_scale_manager and opt

Merge pull request !6036 from liuyang/loss_scale_opt
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
0d1dd6cb42
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/model.py

+ 1
- 1
mindspore/train/model.py View File

@@ -147,7 +147,7 @@ class Model:
def _build_train_network(self):
"""Build train network"""
network = self._network
if self._loss_scale_manager_set and not self._optimizer:
if self._loss_scale_manager is not None and self._optimizer is None:
raise ValueError("Optimizer can not be None when set loss_scale_manager.")
if self._optimizer:
if self._loss_scale_manager_set:


Loading…
Cancel
Save