jongseok kim

Below may work? refer norm stats from https://github.com/openai/CLIP/blob/main/clip/clip.py#L82 ``` class CLIPLoss(torch.nn.Module): def __init__(self, opts): super(CLIPLoss, self).__init__() self.model, self.preprocess = clip.load("ViT-B/32", device="cuda") self.face_pool = torch.nn.AdaptiveAvgPool2d((224, 224)) self.mean = torch.tensor([0.48145466, 0.4578275, 0.40821073],...