Allow MixIT to return non-reduced loss over batch.

asteroid-team / asteroid

The PyTorch-based audio source separation toolkit for researchers

https://asteroid-team.github.io/

MIT License

2.26k stars 422 forks source link

Closed actuallyaswin closed 2 years ago

mpariente commented 3 years ago

Thanks for the PR.

Could you lint the PR and add a test that the expected shape is right please?