Fix device error when using cuda (#4)

- Fix device error when using cuda (030c843a1bb758298a3f0bc6f2564a26aaff878e)


Co-authored-by: Ma Jinyu <[email protected]>

models/tag2text.py

@@ -152,8 +152,7 @@ class RAM(nn.Module):
             self.class_threshold[key] = value
 
     def load_tag_list(self, tag_list_file):
-        with open(tag_list_file, 'r', encoding="utf-8") as f:
-        # with open(tag_list_file, 'r') as f:
+        with open(tag_list_file, 'r', encoding="utf8") as f:
             tag_list = f.read().splitlines()
         tag_list = np.array(tag_list)
         return tag_list
@@ -362,7 +361,7 @@ class Tag2Text_Caption(nn.Module):
             logits = self.fc(tagging_embed[0])
 
             targets = torch.where(
-                torch.sigmoid(logits) > self.class_threshold,
+                torch.sigmoid(logits) > self.class_threshold.to(image.device),
                 torch.tensor(1.0).to(image.device),
                 torch.zeros(self.num_class).to(image.device))