From e4ab67ec76fa3482a466484ca854e76ae67e1966 Mon Sep 17 00:00:00 2001
From: Wei Luning <weiluning@huawei.com>
Date: Thu, 30 Jul 2020 10:26:39 +0800
Subject: [PATCH] fix eval in amp
---
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 fb0eeffcd5..6c79eb7679 100755
--- a/mindspore/train/model.py
+++ b/mindspore/train/model.py
@@ -174,7 +174,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 ["O0", "O3"])
+ self._eval_network = nn.WithEvalCell(self._network, self._loss_fn, self._amp_level in ["O2", "O3"])
self._eval_indexes = [0, 1, 2]
if self._parallel_mode in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL):