Browse Source

loss_scale_opt

tags/v1.0.0
liuyang_655 5 years ago
parent
commit
65317ebccb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/model.py

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

@@ -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:


Loading…
Cancel
Save