minor fixes

This commit is contained in:
shahules786 2022-11-15 21:51:06 +05:30
parent 191c6a7499
commit 434b44ddc9
2 changed files with 18 additions and 1 deletions

View File

@ -1,6 +1,8 @@
import math import math
import multiprocessing import multiprocessing
import os import os
import sys
import warnings
from pathlib import Path from pathlib import Path
from typing import Optional from typing import Optional
@ -80,6 +82,21 @@ class TaskDataset(pl.LightningDataModule):
self._validation = [] self._validation = []
if num_workers is None: if num_workers is None:
num_workers = multiprocessing.cpu_count() // 2 num_workers = multiprocessing.cpu_count() // 2
if num_workers is None:
num_workers = multiprocessing.cpu_count() // 2
if (
num_workers > 0
and sys.platform == "darwin"
and sys.version_info[0] >= 3
and sys.version_info[1] >= 8
):
warnings.warn(
"num_workers > 0 is not supported with macOS and Python 3.8+: "
"setting num_workers = 0."
)
num_workers = 0
self.num_workers = num_workers self.num_workers = num_workers
if min_valid_minutes > 0.0: if min_valid_minutes > 0.0:
self.min_valid_minutes = min_valid_minutes self.min_valid_minutes = min_valid_minutes

View File

@ -93,7 +93,7 @@ class Fileprocessor:
def from_name(cls, name: str, clean_dir, noisy_dir, matching_function=None): def from_name(cls, name: str, clean_dir, noisy_dir, matching_function=None):
if matching_function is None: if matching_function is None:
if name.lower() == "vctk": if name.lower() in ("vctk", "valentini"):
return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_one) return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_one)
elif name.lower() == "dns-2020": elif name.lower() == "dns-2020":
return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_many) return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_many)