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

rename loss

This commit is contained in:
shahules786 2022-10-26 10:27:23 +05:30
parent 04782ba6e9
commit 24a06ba9be
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/loss.py
View File

@ -66,7 +66,7 @@ class Si_SDR:
"Invalid reduction, valid options are sum, mean, None" "Invalid reduction, valid options are sum, mean, None"
) )
self.higher_better = False self.higher_better = False
self.name = "Si-SDR" self.name = "si-sdr"
def __call__(self, prediction: torch.Tensor, target: torch.Tensor): def __call__(self, prediction: torch.Tensor, target: torch.Tensor):