!6036 Repair loss_scale_manager and opt

Merge pull request !6036 from liuyang/loss_scale_opt
pull/6036/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 0d1dd6cb42

@ -147,7 +147,7 @@ class Model:
def _build_train_network(self): def _build_train_network(self):
"""Build train network""" """Build train network"""
network = self._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.") raise ValueError("Optimizer can not be None when set loss_scale_manager.")
if self._optimizer: if self._optimizer:
if self._loss_scale_manager_set: if self._loss_scale_manager_set:

Loading…
Cancel
Save