debug

utils.py

@@ -236,7 +236,11 @@ class ReactionPredictionModel():
         for i, batch in tqdm(enumerate(test_loader), total=len(test_loader), desc="Evaluating"):
             with torch.no_grad():
                 generation_prompts = batch['generation_prompts'][0]
-                inputs = self.tokenizer(generation_prompts, return_tensors="pt", padding=True, truncation=True).to(self.retro_model.device)
+                inputs = self.tokenizer(generation_prompts, return_tensors="pt", padding=True, truncation=True)
+                inputs = {k: v.to(self.retro_model.device) for k, v in inputs.items()}
+                print(inputs)
+                print(self.forward_model.device)
+                print(self.retro_model.device)
                 del inputs['token_type_ids']
                 if task_type == "retrosynthesis":
                     outputs = self.retro_model.generate(**inputs, max_length=512, num_return_sequences=10,