1 parent 72fede0 commit fb65d4dCopy full SHA for fb65d4d
1 file changed
ignite/metrics/gan/fid.py
@@ -31,9 +31,8 @@ def fid_score(
31
except ImportError:
32
raise ModuleNotFoundError("fid_score requires scipy to be installed.")
33
34
- mu1, mu2 = mu1.cpu(), mu2.cpu()
35
- sigma1, sigma2 = sigma1.cpu(), sigma2.cpu()
36
-
+ mu1, mu2 = mu1.detach().cpu(), mu2.detach().cpu()
+ sigma1, sigma2 = sigma1.detach().cpu(), sigma2.detach().cpu()
37
diff = mu1 - mu2
38
39
# Product might be almost singular
0 commit comments