Skip to content

Commit 5a5b0fc

Browse files
nateanlfacebook-github-bot
authored andcommitted
Update description of Squim pipelines (#3254)
Summary: - Add citations of [`TorchAudio-Squim`](https://arxiv.org/abs/2304.01448) publication. - Update descriptions in the `SQUIM_OBJECTIVE` and `SQUIM_SUBJECTIVE` pipelines. Pull Request resolved: #3254 Reviewed By: hwangjeff Differential Revision: D44802015 Pulled By: nateanl fbshipit-source-id: ca08298ec1eafefdd671ff2e010ef18f7372f9f8
1 parent a660271 commit 5a5b0fc

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

docs/source/refs.bib

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -556,3 +556,9 @@ @article{mysore2014can
556556
year={2014},
557557
publisher={IEEE}
558558
}
559+
@article{kumar2023torchaudio,
560+
title={TorchAudio-Squim: Reference-less Speech Quality and Intelligibility measures in TorchAudio},
561+
author={Kumar, Anurag and Tan, Ke and Ni, Zhaoheng and Manocha, Pranay and Zhang, Xiaohui and Henderson, Ethan and Xu, Buye},
562+
journal={arXiv preprint arXiv:2304.01448},
563+
year={2023}
564+
}

torchaudio/prototype/pipelines/squim_pipeline.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,8 @@ def sample_rate(self):
7979
"squim_objective_dns2020.pth",
8080
_sample_rate=16000,
8181
)
82-
SQUIM_OBJECTIVE.__doc__ = """SquimObjective pipeline, trained on the *DNS 2020 Dataset*
83-
:cite:`reddy2020interspeech`.
82+
SQUIM_OBJECTIVE.__doc__ = """SquimObjective pipeline trained using approach described in
83+
:cite:`kumar2023torchaudio` on the *DNS 2020 Dataset* :cite:`reddy2020interspeech`.
8484
8585
The underlying model is constructed by :py:func:`torchaudio.prototype.models.squim_objective_base`.
8686
The weights are under `Creative Commons Attribution 4.0 International License
@@ -166,8 +166,9 @@ def sample_rate(self):
166166
"squim_subjective_bvcc_daps.pth",
167167
_sample_rate=16000,
168168
)
169-
SQUIM_SUBJECTIVE.__doc__ = """SquimSubjective pipeline, trained on the *BVCC*
170-
:cite:`cooper2021voices` and *DAPS* :cite:`mysore2014can` datasets.
169+
SQUIM_SUBJECTIVE.__doc__ = """SquimSubjective pipeline trained
170+
as described in :cite:`manocha2022speech` and :cite:`kumar2023torchaudio`
171+
on the *BVCC* :cite:`cooper2021voices` and *DAPS* :cite:`mysore2014can` datasets.
171172
172173
The underlying model is constructed by :py:func:`torchaudio.prototype.models.squim_subjective_base`.
173174
The weights are under `Creative Commons Attribution Non Commercial 4.0 International

0 commit comments

Comments
 (0)