Commit 88a8dd4d authored by nateanl's avatar nateanl Committed by Zhaoheng Ni
Browse files

Update description of HDemucs pipelines (#2774)

Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/2774

Reviewed By: carolineechen

Differential Revision: D40445274

Pulled By: nateanl

fbshipit-source-id: 6388323a5fa5c548a86829cb3f7cafee5382d18d
parent b703a631
...@@ -82,8 +82,9 @@ HDEMUCS_HIGH_MUSDB_PLUS = SourceSeparationBundle( ...@@ -82,8 +82,9 @@ HDEMUCS_HIGH_MUSDB_PLUS = SourceSeparationBundle(
_sample_rate=44100, _sample_rate=44100,
) )
HDEMUCS_HIGH_MUSDB_PLUS.__doc__ = """Pre-trained music source separation pipeline with HDEMUCS_HIGH_MUSDB_PLUS.__doc__ = """Pre-trained music source separation pipeline with
*Hybrid Demucs* :cite:`defossez2021hybrid` trained on MUSDB-HQ :cite:`MUSDB18HQ` *Hybrid Demucs* :cite:`defossez2021hybrid` trained on both training and test sets of
and additional internal training data. MUSDB-HQ :cite:`MUSDB18HQ` and an additional 150 extra songs from an internal database
that was specifically produced for Meta.
The model is constructed by :func:`~torchaudio.models.hdemucs_high`. The model is constructed by :func:`~torchaudio.models.hdemucs_high`.
...@@ -99,7 +100,7 @@ HDEMUCS_HIGH_MUSDB = SourceSeparationBundle( ...@@ -99,7 +100,7 @@ HDEMUCS_HIGH_MUSDB = SourceSeparationBundle(
_sample_rate=44100, _sample_rate=44100,
) )
HDEMUCS_HIGH_MUSDB.__doc__ = """Pre-trained music source separation pipeline with HDEMUCS_HIGH_MUSDB.__doc__ = """Pre-trained music source separation pipeline with
*Hybrid Demucs* :cite:`defossez2021hybrid` trained on MUSDB-HQ :cite:`MUSDB18HQ`. *Hybrid Demucs* :cite:`defossez2021hybrid` trained on the training set of MUSDB-HQ :cite:`MUSDB18HQ`.
The model is constructed by :func:`~torchaudio.models.hdemucs_high`. The model is constructed by :func:`~torchaudio.models.hdemucs_high`.
Training was performed in the original HDemucs repository `here <https://github.com/facebookresearch/demucs/>`__. Training was performed in the original HDemucs repository `here <https://github.com/facebookresearch/demucs/>`__.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment