File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -82,8 +82,9 @@ def get_model(self) -> torch.nn.Module:
8282 _sample_rate = 44100 ,
8383)
8484HDEMUCS_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
8889The 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)
101102HDEMUCS_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
104105The model is constructed by :func:`~torchaudio.models.hdemucs_high`.
105106Training was performed in the original HDemucs repository `here <https://github.com/facebookresearch/demucs/>`__.
You can’t perform that action at this time.
0 commit comments