Disable flaky test in test_amp_conversion (#16031)

apache · Aug 29, 2019 · 649429d · 649429d
1 parent 9b906a5
commit 649429d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/python/gpu/test_contrib_amp.py b/tests/python/gpu/test_contrib_amp.py
@@ -329,10 +329,13 @@ def check_amp_convert_bucketing_module():
                            batch_end_callback=mx.callback.Speedometer(batch_size, 1))
 
         # AMP conversion with cast_optional_params set to true
+        # Flaky test when cast_optional_params set to True : https://github.com/apache/incubator-mxnet/issues/16030
+        '''
         result_model = amp.convert_bucketing_module(model, cast_optional_params=True)
         result_model.bind(data_val.provide_data, data_val.provide_label, for_training=False)
         result_model.score(data_val, mx.metric.Perplexity(invalid_label),
                            batch_end_callback=mx.callback.Speedometer(batch_size, 1))
+        '''
 
 
     with mx.Context(mx.gpu(0)):