diff --git a/freqtrade/freqai/torch/PyTorchModelTrainer.py b/freqtrade/freqai/torch/PyTorchModelTrainer.py
index e7c4d53be..2b0090c78 100644
--- a/freqtrade/freqai/torch/PyTorchModelTrainer.py
+++ b/freqtrade/freqai/torch/PyTorchModelTrainer.py
@@ -181,8 +181,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
             "pytrainer": self
         }, path)
 
-    def load(self, path: Path, device: Optional[str] = None):
-        checkpoint = torch.load(path, map_location=device)
+    def load(self, path: Path):
+        checkpoint = torch.load(path)
         return self.load_from_checkpoint(checkpoint)
 
     def load_from_checkpoint(self, checkpoint: Dict):