This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new f7a0025 Fix flaky test
test_operator_gpu.test_spatial_transformer_with_type (#7645) (#11444)
f7a0025 is described below
commit f7a00250cf83ba34779b7c6ffc02532d20795f2f
Author: ddavydenko <[email protected]>
AuthorDate: Thu Jun 28 16:16:29 2018 -0700
Fix flaky test test_operator_gpu.test_spatial_transformer_with_type (#7645)
(#11444)
---
tests/python/gpu/test_operator_gpu.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/tests/python/gpu/test_operator_gpu.py
b/tests/python/gpu/test_operator_gpu.py
index fd76990..3efc284 100644
--- a/tests/python/gpu/test_operator_gpu.py
+++ b/tests/python/gpu/test_operator_gpu.py
@@ -701,8 +701,7 @@ def test_grid_generator_with_type():
check_consistency(sym, ctx_list, grad_req="add")
[email protected]("test fails intermittently. temporarily disabled till it gets
fixed. tracked at https://github.com/apache/incubator-mxnet/issues/7645";)
-@with_seed(1234)
+@with_seed()
def test_spatial_transformer_with_type():
data = mx.sym.Variable('data')
loc = mx.sym.Flatten(data)
@@ -711,8 +710,8 @@ def test_spatial_transformer_with_type():
loc = mx.sym.FullyConnected(data=loc, num_hidden=6)
sym = mx.sym.SpatialTransformer(data=data, loc=loc, target_shape=(10, 10),
transform_type="affine",
sampler_type="bilinear")
- ctx_list = [{'ctx': mx.gpu(0), 'data': (1, 5, 10, 10), 'type_dict':
{'data': np.float32}},
- {'ctx': mx.cpu(0), 'data': (1, 5, 10, 10), 'type_dict':
{'data': np.float32}}]
+ ctx_list = [{'ctx': mx.gpu(0), 'data': (1, 5, 10, 10), 'type_dict':
{'data': np.float64}},
+ {'ctx': mx.cpu(0), 'data': (1, 5, 10, 10), 'type_dict':
{'data': np.float64}}]
check_consistency(sym, ctx_list)
check_consistency(sym, ctx_list, grad_req="add")
