Removing unecessary code that transforms to cpu, while it is on cpu

Author: ArlindKadra

autoPyTorch/pipeline/components/setup/network/base_network.py

@@ -113,7 +113,7 @@ def predict(self, loader: torch.utils.data.DataLoader) -> torch.Tensor:
         """
         if len(self.network_snapshots) == 0:
             assert self.network is not None
-            return self._predict(network=self.network, loader=loader).cpu().numpy()
+            return self._predict(network=self.network, loader=loader).numpy()
         else:
             # if there are network snapshots,
             # take average of predictions of all snapshots