Increase number of attempts for BiDAF training test, which is very flaky
@@ -42,7 +42,9 @@ class BidirectionalAttentionFlowTest(ModelTestCase):
assert span_end < self.dataset.instances[0].fields['passage'].sequence_length()
assert isinstance(output_dict['best_span_str'][0], str)
- @flaky
+ # Some recent efficiency changes (using bmm for `weighted_sum`, the more efficient
+ # `masked_softmax`...) have made this _very_ flaky...
+ @flaky(max_runs=5)
def test_model_can_train_save_and_load(self):
self.ensure_model_can_train_save_and_load(self.param_file)
