From c7b796df778c49504454ba07fe6fa6fd060a8352 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 5 Sep 2023 20:27:00 +0800 Subject: [PATCH] Update bert_gen.py --- bert_gen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bert_gen.py b/bert_gen.py index c3a3234..265d091 100644 --- a/bert_gen.py +++ b/bert_gen.py @@ -14,7 +14,7 @@ def process_line(line): if torch.cuda.is_available(): gpu_id = rank % torch.cuda.device_count() device = torch.device(f"cuda:{gpu_id}") - logger.info(f"Rank {rank} uses device {device}") + #logger.info(f"Rank {rank} uses device {device}") wav_path, _, language_str, text, phones, tone, word2ph = line.strip().split("|") phone = phones.split(" ") tone = [int(i) for i in tone.split(" ")]