Update runner.py to load models to correct device

rsl_rl/runners/runner.py

@@ -365,7 +365,7 @@ def _update(self) -> None:
 
     def load(self, path: str) -> Any:
         """Restores the agent and runner state from a file."""
-        content = torch.load(path)
+        content = torch.load(path, map_location=self.device)
 
         assert "agent" in content
         assert "data" in content