minor fixes
This commit is contained in:
parent
191c6a7499
commit
434b44ddc9
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue