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(" ")]