debug

nnet/xvector.py

@@ -952,7 +952,7 @@ def xtrain(speaker_number,
     logging.critical(f"Use {num_thread} cpus")
     logging.critical(f"Start process at {time.strftime('%H:%M:%S', time.localtime())}")
 
-    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     # Start from scratch
     if model_name is None and model_yaml in ["xvector", "rawnet2", "resnet34", "fastresnet34"]:
         # Initialize a first model
@@ -1521,7 +1521,7 @@ def xtraintop(model_filename,
     :param num_thread:
     :return:
     """
-    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     # Load the model
     if isinstance(model_filename, str):