tighten test_sampled_equals_unsampled_when_biased_against_non_sampled…

…_positions bound (#5549) Co-authored-by: Akshita Bhagia <akshita23bhagia@gmail.com>
allenai · Feb 10, 2022 · 3c2299a · 3c2299a
1 parent e463084
commit 3c2299a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/modules/sampled_softmax_loss_test.py b/tests/modules/sampled_softmax_loss_test.py
@@ -95,4 +95,4 @@ def fake_choice(num_words: int, num_samples: int) -> Tuple[np.ndarray, int]:
         # Should be close
 
         pct_error = (sampled_loss - full_loss) / full_loss
-        assert abs(pct_error) < 0.001
+        assert abs(pct_error) < 0.0003