未验证 提交 a37be82f 编写于 作者: S Sing_chan 提交者: GitHub

modify according to zhouwei's comment (#38166)

上级 30973183
...@@ -632,7 +632,8 @@ class TestImperativeLambOptimizer(TestImperativeOptimizerBase): ...@@ -632,7 +632,8 @@ class TestImperativeLambOptimizer(TestImperativeOptimizerBase):
learning_rate=0.002, exclude_from_weight_decay_fn=exclude_fn) learning_rate=0.002, exclude_from_weight_decay_fn=exclude_fn)
return optimizer return optimizer
def test_lamb(self): # should fix: may fail in CI-windows
def _test_lamb(self):
self._check_mlp() self._check_mlp()
......
...@@ -721,7 +721,8 @@ class TestImperativeLambOptimizer(TestImperativeOptimizerBase): ...@@ -721,7 +721,8 @@ class TestImperativeLambOptimizer(TestImperativeOptimizerBase):
learning_rate=0.002, exclude_from_weight_decay_fn=exclude_fn) learning_rate=0.002, exclude_from_weight_decay_fn=exclude_fn)
return optimizer return optimizer
def test_lamb(self): # should fix: may fail in CI-windows
def _test_lamb(self):
self._check_mlp() self._check_mlp()
......
...@@ -95,7 +95,6 @@ disable_wingpu_test="^test_model$|\ ...@@ -95,7 +95,6 @@ disable_wingpu_test="^test_model$|\
^test_activation_op$|\ ^test_activation_op$|\
^test_norm_nn_grad$|\ ^test_norm_nn_grad$|\
^test_bilinear_interp_op$|\ ^test_bilinear_interp_op$|\
^test_imperative_optimizer_v2$|\
^disable_wingpu_test$" ^disable_wingpu_test$"
# /*==================Fixed Disabled Windows GPU MKL unittests==============================*/ # /*==================Fixed Disabled Windows GPU MKL unittests==============================*/
...@@ -188,7 +187,6 @@ long_time_test="^test_gru_op$|\ ...@@ -188,7 +187,6 @@ long_time_test="^test_gru_op$|\
^test_sgd_op$|\ ^test_sgd_op$|\
^test_transformer$|\ ^test_transformer$|\
^test_imperative_auto_mixed_precision$|\ ^test_imperative_auto_mixed_precision$|\
^test_imperative_optimizer_v2$|\
^test_trt_matmul_quant_dequant$|\ ^test_trt_matmul_quant_dequant$|\
^test_strided_slice_op$" ^test_strided_slice_op$"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册