未验证 提交 69cfb7a2 编写于 作者: 0 0x45f 提交者: GitHub

[Dy2St]Skip windows for test_mnist_pure_fp16 (#38214)

上级 0cc5e22c
...@@ -12,6 +12,7 @@ ...@@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import os
import paddle import paddle
import unittest import unittest
import numpy as np import numpy as np
...@@ -32,7 +33,7 @@ class TestPureFP16(TestMNIST): ...@@ -32,7 +33,7 @@ class TestPureFP16(TestMNIST):
return self.train(to_static=False) return self.train(to_static=False)
def test_mnist_to_static(self): def test_mnist_to_static(self):
if paddle.fluid.is_compiled_with_cuda(): if paddle.fluid.is_compiled_with_cuda() and os.name != 'nt':
dygraph_loss = self.train_dygraph() dygraph_loss = self.train_dygraph()
static_loss = self.train_static() static_loss = self.train_static()
# NOTE: In pure fp16 training, loss is not stable, so we enlarge atol here. # NOTE: In pure fp16 training, loss is not stable, so we enlarge atol here.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册