fix model bug

2toinf · Mar 19, 2024 · 599860a · 599860a
1 parent cffb75d
commit 599860a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/DecisionNCE/model.py b/DecisionNCE/model.py
@@ -17,6 +17,7 @@ def __init__(self, modelid="RN50", device="cuda"):
         # CLIP precision
         model.float()
         self.model = model 
+        del self.model.logit_scale
         self.model.train()
         self.transforms = cliptransforms
         self.transforms_tensor = nn.Sequential(