🐛 fix device discrepance in evaluator

mowl/evaluation/base.py

@@ -48,8 +48,8 @@ def __init__(self, dataset, device="cpu", batch_size=16):
         eval_heads, eval_tails = self.dataset.evaluation_classes
 
         print(f"Number of evaluation classes: {len(eval_heads)}")
-        self.evaluation_heads = th.tensor([self.class_to_id[c] for c in eval_heads.as_str], dtype=th.long)
-        self.evaluation_tails = th.tensor([self.class_to_id[c] for c in eval_tails.as_str], dtype=th.long)
+        self.evaluation_heads = th.tensor([self.class_to_id[c] for c in eval_heads.as_str], dtype=th.long).to(self.device)
+        self.evaluation_tails = th.tensor([self.class_to_id[c] for c in eval_tails.as_str], dtype=th.long).to(self.device)
 
 
     @property
@@ -71,6 +71,7 @@ def evaluate_base(self, model, eval_tuples, mode="test",
                       filter_deductive_closure=False,
                       **kwargs):
 
+        model = model.to(self.device)
         num_heads, num_tails = len(self.evaluation_heads), len(self.evaluation_tails)
         model.eval()
         if not mode in ["valid", "test"]: