debug model loading

src/model.py

@@ -101,7 +101,9 @@ class Tox21SNNClassifier(nn.Module):
         return x  # x.view(x.size(0), self.num_tasks)
 
     def load_model(self, path: str):
-        state_dict = torch.load(path, weights_only=False)["model"]
+        state_dict = torch.load(
+            path, weights_only=False, map_location=torch.device("cpu")
+        )["model"]
         self.load_state_dict(state_dict)
         self.eval()