Skip to content

Commit b298167

Browse files
committed
Merge branch 'master' into fix_test_roc_on_cuda
2 parents fc9a6fc + c6b75ee commit b298167

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/ignite/metrics/gan/test_fid.py

Lines changed: 2 additions & 2 deletions
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)