Commit c5253816 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

debug

parent 9bf46eca
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment