From 65317ebccbfdcb2b227d26fcb36e02acabd78a5b Mon Sep 17 00:00:00 2001 From: liuyang_655 Date: Fri, 11 Sep 2020 11:29:00 +0800 Subject: [PATCH] loss_scale_opt --- mindspore/train/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index d67d4fef98..b866d435a0 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -148,7 +148,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: