Skip to content

Commit 88a8dd4

Browse files
committed
Update description of HDemucs pipelines (#2774)
Summary: Pull Request resolved: #2774 Reviewed By: carolineechen Differential Revision: D40445274 Pulled By: nateanl fbshipit-source-id: 6388323a5fa5c548a86829cb3f7cafee5382d18d
1 parent b703a63 commit 88a8dd4

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

torchaudio/pipelines/_source_separation_pipeline.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,8 +82,9 @@ def get_model(self) -> torch.nn.Module:
8282
_sample_rate=44100,
8383
)
8484
HDEMUCS_HIGH_MUSDB_PLUS.__doc__ = """Pre-trained music source separation pipeline with
85-
*Hybrid Demucs* :cite:`defossez2021hybrid` trained on MUSDB-HQ :cite:`MUSDB18HQ`
86-
and additional internal training data.
85+
*Hybrid Demucs* :cite:`defossez2021hybrid` trained on both training and test sets of
86+
MUSDB-HQ :cite:`MUSDB18HQ` and an additional 150 extra songs from an internal database
87+
that was specifically produced for Meta.
8788
8889
The model is constructed by :func:`~torchaudio.models.hdemucs_high`.
8990
@@ -99,7 +100,7 @@ def get_model(self) -> torch.nn.Module:
99100
_sample_rate=44100,
100101
)
101102
HDEMUCS_HIGH_MUSDB.__doc__ = """Pre-trained music source separation pipeline with
102-
*Hybrid Demucs* :cite:`defossez2021hybrid` trained on MUSDB-HQ :cite:`MUSDB18HQ`.
103+
*Hybrid Demucs* :cite:`defossez2021hybrid` trained on the training set of MUSDB-HQ :cite:`MUSDB18HQ`.
103104
104105
The model is constructed by :func:`~torchaudio.models.hdemucs_high`.
105106
Training was performed in the original HDemucs repository `here <https://github.com/facebookresearch/demucs/>`__.

0 commit comments

Comments
 (0)