from torchmetrics.image.fid import FrechetInceptionDistance