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

add documentation

This commit is contained in:
shahules786 2022-11-02 18:00:05 +05:30
parent 7e298b811f
commit b857754626
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/models/model.py
View File

@ -37,7 +37,7 @@ class Model(pl.LightningModule):
Enhancer dataset used for training/validation
duration: float, optional
duration used for training/inference
loss : string or List of strings, default to "mse"
loss : string or List of strings or custom loss (nn.Module), default to "mse"
loss functions to be used. Available ("mse","mae","Si-SDR")
"""