fix device issue: cpu

src/er_client/sentence_selection.py

@@ -23,7 +23,8 @@ class SentSelector:
         self.bert_model = BertForSequenceEncoder.from_pretrained(pretrained_bert_path)
 
         self.rank_model = inference_model(self.bert_model, self.args)
-        self.rank_model.load_state_dict(torch.load(select_model_path)['model'])
+        self.rank_model.load_state_dict(torch.load(select_model_path,
+                                                   map_location=None if self.use_cuda else torch.device('cpu'))['model'])
 
         if self.use_cuda:
             self.bert_model = self.bert_model.cuda()