| """Prepare the exact common metric weights before offline evaluation workers.""" | |
| from pathlib import Path | |
| import sys | |
| sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "shared/metric_vendor")) | |
| from torchmetrics.image.fid import FrechetInceptionDistance | |
| from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity | |
| LearnedPerceptualImagePatchSimilarity(net_type="alex", normalize=True).eval() | |
| FrechetInceptionDistance(feature=2048, normalize=False).eval() | |
| print("Common LPIPS AlexNet and FID Inception weights are cached.", flush=True) | |