Increase tolerance to the default.
@@ -48,7 +48,7 @@ class BidirectionalAttentionFlowTest(ModelTestCase):
# `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, tolerance=2e-5)
+ self.ensure_model_can_train_save_and_load(self.param_file, tolerance=1e-4)
@flaky
def test_batch_predictions_are_consistent(self):
