Update ignite/metrics/gan/fid.py

Co-authored-by: Copilot <[email protected]>

Author: goanpeca

ignite/metrics/gan/fid.py

@@ -37,7 +37,7 @@ def fid_score(
     diff = mu1 - mu2
 
     # Product might be almost singular
-    covmean, _ = scipy.linalg.sqrtm(sigma1.mm(sigma2).numpy(), disp=False)
+    covmean, _ = scipy.linalg.sqrtm(sigma1.mm(sigma2).detach().cpu().numpy(), disp=False)
     # Numerical error might give slight imaginary component
     if np.iscomplexobj(covmean):
         if not np.allclose(np.diagonal(covmean).imag, 0, atol=1e-3):