Update networks.py

networks.py

@@ -35,6 +35,7 @@ class SpeechModel:
         - args: Argument parser object containing settings like whether to use CUDA (GPU) or not.
         """
         # Check if a GPU is available
+        """
         if torch.cuda.is_available():
             # Find the GPU with the most free memory using a custom method
             free_gpu_id = self.get_free_gpu()
@@ -52,7 +53,15 @@ class SpeechModel:
             # If no GPU is detected, use the CPU
             args.use_cuda = 0
             self.device = torch.device('cpu')
-
+        """    
+        if torch.cuda.is_available():
+            print('GPU is found and used!')
+            self.device = torch.device('cuda')
+        else:
+            # If no GPU is detected, use the CPU
+            args.use_cuda = 0
+            self.device = torch.device('cpu')
+            
         self.args = args
         self.model = None
         self.name = None