import torch
_ = torch.manual_seed(123)
from torcheval.metrics import FrechetInceptionDistance
fid = FrechetInceptionDistance()
# generate two slightly overlapping image intensity distributions
imgs_dist1 = torch.randint(0, 200, (100, 3, 299, 299), )/255
imgs_dist2 = torch.randint(100, 255, (100, 3, 299, 299))/255
fid.update(imgs_dist1, is_real=True)
fid.update(imgs_dist1+.1, is_real=False)
print(fid.compute())