Skip to content

Instantly share code, notes, and snippets.

@adambielski
adambielski / callback_fid.py
Created November 3, 2020 23:20
FID callback for PyTorch Lightning
''' fid code and inception model from https://github.com/mseitzer/pytorch-fid '''
import pytorch_lightning as pl
from pytorch_lightning.utilities import rank_zero_only
from scipy import linalg
from .inception import InceptionV3 # https://github.com/mseitzer/pytorch-fid
import pickle
import torch
import numpy as np
from tqdm import tqdm