magenta / mt3

MT3: Multi-Task Multitrack Music Transcription
Apache License 2.0
1.43k stars 187 forks source link

How do I train mt3 locally #162

Open opria123 opened 1 month ago

opria123 commented 1 month ago

I want to train a model using mt3 on my local PC, however I am running into quite a few different issues. I tried to follow the t5x documentation on training that is linked to in the readme but they only have instruction on how to train the model in GCP with TPUs. I tried running the setup.py file with install but I am getting this error:


C:\ProgramData\anaconda3\Lib\site-packages\setuptools\__init__.py:80: _DeprecatedInstaller: setuptools.installer and fetch_build_eggs are deprecated.
!!

        ********************************************************************************
        Requirements should be satisfied by a PEP 517 installer.
        If you are using pip, you can try `pip install --use-pep517`.
        ********************************************************************************

!!
  dist.fetch_build_eggs(dist.setup_requires)
running install
C:\ProgramData\anaconda3\Lib\site-packages\setuptools\_distutils\cmd.py:66: SetuptoolsDeprecationWarning: setup.py install is deprecated.
!!

        ********************************************************************************
        Please avoid running ``setup.py`` directly.
        Instead, use pypa/build, pypa/installer or other
        standards-based tools.

        See https://blog.ganssle.io/articles/2021/10/setup-py-deprecated.html for details.
        ********************************************************************************

!!
  self.initialize_options()
C:\ProgramData\anaconda3\Lib\site-packages\setuptools\_distutils\cmd.py:66: EasyInstallDeprecationWarning: easy_install command is deprecated.
!!

        ********************************************************************************
        Please avoid running ``setup.py`` and ``easy_install``.
        Instead, use pypa/build, pypa/installer or other
        standards-based tools.

        See https://github.com/pypa/setuptools/issues/917 for details.
        ********************************************************************************

!!
  self.initialize_options()
running bdist_egg
running egg_info
writing mt3.egg-info\PKG-INFO
writing dependency_links to mt3.egg-info\dependency_links.txt
writing requirements to mt3.egg-info\requires.txt
writing top-level names to mt3.egg-info\top_level.txt
reading manifest file 'mt3.egg-info\SOURCES.txt'
adding license file 'LICENSE'
writing manifest file 'mt3.egg-info\SOURCES.txt'
installing library code to build\bdist.win-amd64\egg
running install_lib
running build_py
creating build\bdist.win-amd64\egg
creating build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\datasets.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\event_codec.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\event_codec_test.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\inference.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\layers.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\layers_test.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\metrics.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\metrics_utils.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\metrics_utils_test.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\mixing.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\models.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\network.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\note_sequences.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\note_sequences_test.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\preprocessors.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\run_length_encoding.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\run_length_encoding_test.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\spectral_ops.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\spectrograms.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\summaries.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\tasks.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\version.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\vocabularies.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\vocabularies_test.py -> build\bdist.win-amd64\egg\mt3
copying build\lib\mt3\__init__.py -> build\bdist.win-amd64\egg\mt3
byte-compiling build\bdist.win-amd64\egg\mt3\datasets.py to datasets.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\event_codec.py to event_codec.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\event_codec_test.py to event_codec_test.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\inference.py to inference.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\layers.py to layers.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\layers_test.py to layers_test.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\metrics.py to metrics.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\metrics_utils.py to metrics_utils.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\metrics_utils_test.py to metrics_utils_test.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\mixing.py to mixing.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\models.py to models.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\network.py to network.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\note_sequences.py to note_sequences.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\note_sequences_test.py to note_sequences_test.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\preprocessors.py to preprocessors.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\run_length_encoding.py to run_length_encoding.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\run_length_encoding_test.py to run_length_encoding_test.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\spectral_ops.py to spectral_ops.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\spectrograms.py to spectrograms.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\summaries.py to summaries.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\tasks.py to tasks.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\version.py to version.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\vocabularies.py to vocabularies.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\vocabularies_test.py to vocabularies_test.cpython-311.pyc
byte-compiling build\bdist.win-amd64\egg\mt3\__init__.py to __init__.cpython-311.pyc
creating build\bdist.win-amd64\egg\EGG-INFO
copying mt3.egg-info\PKG-INFO -> build\bdist.win-amd64\egg\EGG-INFO
copying mt3.egg-info\SOURCES.txt -> build\bdist.win-amd64\egg\EGG-INFO
copying mt3.egg-info\dependency_links.txt -> build\bdist.win-amd64\egg\EGG-INFO
copying mt3.egg-info\requires.txt -> build\bdist.win-amd64\egg\EGG-INFO
copying mt3.egg-info\top_level.txt -> build\bdist.win-amd64\egg\EGG-INFO
zip_safe flag not set; analyzing archive contents...
creating 'dist\mt3-0.0.1-py3.11.egg' and adding 'build\bdist.win-amd64\egg' to it
removing 'build\bdist.win-amd64\egg' (and everything under it)
Processing mt3-0.0.1-py3.11.egg
Copying mt3-0.0.1-py3.11.egg to c:\programdata\anaconda3\lib\site-packages
Adding mt3 0.0.1 to easy-install.pth file

Installed c:\programdata\anaconda3\lib\site-packages\mt3-0.0.1-py3.11.egg
Processing dependencies for mt3==0.0.1
Searching for t5
Reading https://pypi.org/simple/t5/
Downloading https://files.pythonhosted.org/packages/8f/ea/4efa24fb3706c565c60b2fa2f59a0344df9229db2d014d51e23f1ce03f98/t5-0.9.4-py2.py3-none-any.whl#sha256=53e40efdfc4d8f614cd8d38301e39d5c146f9d5fd66b6823d9c181b60370b2b5
Best match: t5 0.9.4
Processing t5-0.9.4-py2.py3-none-any.whl
Installing t5-0.9.4-py2.py3-none-any.whl to c:\programdata\anaconda3\lib\site-packages
Adding t5 0.9.4 to easy-install.pth file
detected new path './mt3-0.0.1-py3.11.egg'
Installing t5_cache_tasks-script.py script to C:\ProgramData\anaconda3\Scripts
Installing t5_cache_tasks.exe script to C:\ProgramData\anaconda3\Scripts
Installing t5_inspect_tasks-script.py script to C:\ProgramData\anaconda3\Scripts
Installing t5_inspect_tasks.exe script to C:\ProgramData\anaconda3\Scripts
Installing t5_mesh_transformer-script.py script to C:\ProgramData\anaconda3\Scripts
Installing t5_mesh_transformer.exe script to C:\ProgramData\anaconda3\Scripts

Installed c:\programdata\anaconda3\lib\site-packages\t5-0.9.4-py3.11.egg
Searching for note-seq@ git+https://github.com/magenta/note-seq#egg=note_seq
Reading https://pypi.org/simple/note-seq/
Downloading https://files.pythonhosted.org/packages/01/f5/7f15207a89b6968f3ba00f69ed75cfcda49d8921736660e73c670ec1510d/note_seq-0.0.5-py3-none-any.whl#sha256=6f47066641c277effd4cc88b93154b3eb45fd4d647760a2338b58ce82af87eb9
Best match: note-seq 0.0.5
Processing note_seq-0.0.5-py3-none-any.whl
Installing note_seq-0.0.5-py3-none-any.whl to c:\programdata\anaconda3\lib\site-packages
Adding note-seq 0.0.5 to easy-install.pth file
detected new path './t5-0.9.4-py3.11.egg'

Installed c:\programdata\anaconda3\lib\site-packages\note_seq-0.0.5-py3.11.egg
Searching for airio@ git+https://github.com/google/airio#egg=airio
Reading https://pypi.org/simple/airio/
Couldn't find index page for 'airio' (maybe misspelled?)
Scanning index of all packages (this may take a while)
Reading https://pypi.org/simple/
No local packages or working download links found for airio@ git+https://github.com/google/airio#egg=airio
error: Could not find suitable distribution for Requirement.parse('airio@ git+https://github.com/google/airio#egg=airio')```