maum-ai / voicefilter

Unofficial PyTorch implementation of Google AI's VoiceFilter system

Home Page:http://swpark.me/voicefilter

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Question about utils/evaluation.py

lawlict opened this issue · comments

Hello @seungwonpark , thank you greatly for your work!
I notice that the utils/evalluation.py has a "break" in the loop of test dataloader.
That is, in the evaluation process, only the first case generated from test dataloader is taken account into computing test loss and test SDR. Could this raise problems like #5 and #9 ?

Look forward to your response.

Hi, @lawlict

Yes, our current SDR calculation process may cause SDR value to look strange since it evaluates only once.

commented

@lawlict from dataloader.py you can get the batchsize of testloader have only 1, so evalluation.py has a "break" may be ok