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 multiprocessing
import os
import sys
import warnings
from pathlib import Path
from typing import Optional
@ -80,6 +82,21 @@ class TaskDataset(pl.LightningDataModule):
self._validation = []
if num_workers is None:
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
if min_valid_minutes > 0.0:
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):
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)
elif name.lower() == "dns-2020":
return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_many)