Commit 63f9ccf6 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

merge

parents 552805f1 19953b86
......@@ -651,11 +651,9 @@ def xtrain(speaker_number,
num_thread = multiprocessing.cpu_count()
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")
# Start from scratch
if model_name is None and model_yaml in ["xvector", "rawnet2"]:
# Initialize a first model
......
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