Debug rank GPU

svc-develop-team · Jul 23, 2023 · 174cb33 · 174cb33
1 parent 3ab6f1f
commit 174cb33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/preprocess_hubert_f0.py b/preprocess_hubert_f0.py
@@ -114,7 +114,7 @@ def process_batch(file_chunk, f0p, diff=False, mel_extractor=None):
     hmodel = utils.get_speech_encoder(speech_encoder, device=device)
     logger.info(f"Loaded speech encoder for rank {rank}")
     for filename in tqdm(file_chunk):
-        process_one(filename, hmodel, f0p, rank, diff, mel_extractor)
+        process_one(filename, hmodel, f0p, gpu_id, diff, mel_extractor)
 
 def parallel_process(filenames, num_processes, f0p, diff, mel_extractor):
     with ProcessPoolExecutor(max_workers=num_processes) as executor: