Merge branch 'colab' into dev
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
import argparse
|
||||
import os
|
||||
import sys
|
||||
from multiprocessing import Pool, cpu_count
|
||||
|
||||
import librosa
|
||||
@@ -10,6 +9,7 @@ from tqdm import tqdm
|
||||
|
||||
from config import config
|
||||
from tools.log import logger
|
||||
from tools.stdout_wrapper import SAFE_STDOUT
|
||||
|
||||
|
||||
def normalize_audio(data, sr):
|
||||
@@ -97,7 +97,7 @@ if __name__ == "__main__":
|
||||
|
||||
pool = Pool(processes=processes)
|
||||
for _ in tqdm(
|
||||
pool.imap_unordered(process, tasks), file=sys.stdout, total=len(tasks)
|
||||
pool.imap_unordered(process, tasks), file=SAFE_STDOUT, total=len(tasks)
|
||||
):
|
||||
pass
|
||||
|
||||
|
||||
Reference in New Issue
Block a user