From 290ac5cb1558745e56ea8cbcfcf8a9b8ba37182c Mon Sep 17 00:00:00 2001 From: Jong Wook Kim Date: Mon, 22 Mar 2021 22:09:57 -0400 Subject: [PATCH] Correctly initializing the logit scale parameter adding numpy import --- clip/model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/clip/model.py b/clip/model.py index 2a09245..9cf262a 100644 --- a/clip/model.py +++ b/clip/model.py @@ -1,6 +1,7 @@ from collections import OrderedDict from typing import Tuple, Union +import numpy as np import torch import torch.nn.functional as F from torch import nn