Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dataset type fix

This commit is contained in:
shahules786 2022-09-24 12:46:40 +05:30
parent 48d5f9c21e
commit 71b98ba67c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/models/model.py
View File

@ -13,7 +13,7 @@ from pathlib import Path
from enhancer import __version__
from enhancer.data.dataset import Dataset
from enhancer.data.dataset import EnhancerDataset
from enhancer.utils.io import Audio
from enhancer.utils.loss import Avergeloss
from enhancer.inference import Inference
@ -29,7 +29,7 @@ class Model(pl.LightningModule):
num_channels:int=1,
sampling_rate:int=16000,
lr:float=1e-3,
dataset:Optional[Dataset]=None,
dataset:Optional[EnhancerDataset]=None,
duration:Optional[float]=None,
loss: Union[str, List] = "mse",
metric:Union[str,List] = "mse"