Skip to content

Commit 27dc13e

Browse files
authored
Merge branch 'master' into fix_test_roc_on_cuda
2 parents fc9a6fc + c6b75ee commit 27dc13e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/ignite/metrics/gan/test_fid.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -67,9 +67,9 @@ def test_compute_fid_from_features(available_device):
6767
mu1, sigma1 = train_samples.mean(axis=0), cov(train_samples, rowvar=False)
6868
mu2, sigma2 = test_samples.mean(axis=0), cov(test_samples, rowvar=False)
6969

70-
tol = 1e-4 if available_device == "mps" else 1e-5
70+
tol = 5e-4 if available_device == "mps" else 1e-5
7171
assert (
72-
pytest.approx(pytorch_fid_score.calculate_frechet_distance(mu1, sigma1, mu2, sigma2), rel=tol)
72+
pytest.approx(pytorch_fid_score.calculate_frechet_distance(mu1, sigma1, mu2, sigma2), abs=tol)
7373
== fid_scorer.compute()
7474
)
7575

0 commit comments

Comments
 (0)