|
|
@@ -148,7 +148,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: |
|
|
|