Merge pull request #47 from jmisilo/46-temp-check

46 - Temperature checking
jmisilo · Oct 30, 2022 · fb893a0 · fb893a0
2 parents b8ec13b + a91f62a
commit fb893a0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/model/model.py b/src/model/model.py
@@ -157,7 +157,10 @@ def forward(self, img, temperature=1.0):
                 caption: generated caption [str]
                 tokens: generated tokens [torch.Tensor]
         '''
-        # only one image at a time
+
+        if temperature <= 0.0:
+            temperature = 1.0
+            print('Temperature must be positive. Setting it to 1.0')
 
         with torch.no_grad():
             img_embedded = self.ie(img)