enable auto-detection of cuda

tuhlbox/torch_classifier.py

@@ -26,10 +26,13 @@ class TorchClassifier(ClassifierMixin, BaseEstimator):
         batch_size: int = 64,
         max_epochs: int = 5,
         learn_rate: float = 1e-3,
-        device: str = "cuda",
+        device: str = None,
         model_kwargs: Dict[str, Any] = None,
         optimizer: Type[torch.optim.Optimizer] = torch.optim.Adam,
     ):
+        if device is None:
+            device = "cuda" if torch.cuda.is_available() else "cpu"
+
         self.module = module
         self.device = device
         self.batch_size = batch_size