|
|
|
@@ -193,7 +193,7 @@ class Model: |
|
|
|
else: |
|
|
|
if self._loss_fn is None: |
|
|
|
raise ValueError("loss_fn can not be None.") |
|
|
|
self._eval_network = nn.WithEvalCell(self._network, self._loss_fn, self._amp_level in ["O2", "O3"]) |
|
|
|
self._eval_network = nn.WithEvalCell(self._network, self._loss_fn, self._amp_level in ["O2", "O3", "auto"]) |
|
|
|
self._eval_indexes = [0, 1, 2] |
|
|
|
|
|
|
|
if self._parallel_mode in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL): |
|
|
|
|