Imported from https://github.com/mseitzer/pytorch-fid