import fileprocessor

This commit is contained in:
shahules786 2022-09-25 11:52:16 +05:30
parent 4cc96c7b09
commit 2b6b848f0a
1 changed files with 2 additions and 1 deletions

View File

@ -8,9 +8,10 @@ from torch.utils.data import IterableDataset, DataLoader, Dataset
import torch.nn.functional as F import torch.nn.functional as F
from typing import Optional from typing import Optional
from enhancer.data.fileprocessor import Fileprocessor
from enhancer.utils.random import create_unique_rng from enhancer.utils.random import create_unique_rng
from enhancer.utils.io import Audio from enhancer.utils.io import Audio
from enhancer.utils import Fileprocessor, check_files from enhancer.utils import check_files
from enhancer.utils.config import Files from enhancer.utils.config import Files
class TrainDataset(IterableDataset): class TrainDataset(IterableDataset):