zhvng / open-musiclm

Implementation of MusicLM, a text to music model published by Google Research, with a few modifications.
https://arxiv.org/abs/2301.11325
MIT License
522 stars 59 forks source link

can't train the clap #21

Open vivagwb opened 1 year ago

vivagwb commented 1 year ago

(open-musiclm) G:\Learn\AmateurLearning\AI\Practice\open-musiclm-main>python ./scripts/train_clap_rvq.py --results_folder ./results/clap_rvq --model_config ./configs/model/musiclm_small.json --training_config ./configs/training/train_musiclm_fma.json loading clap... Some weights of the model checkpoint at roberta-base were not used when initializing RobertaModel: ['lm_head.decoder.weight', 'lm_head.dense.weight', 'lm_head.bias', 'lm_head.layer_norm.weight', 'lm_head.layer_norm.bias', 'lm_head.dense.bias']

MichaelALong commented 1 year ago

This error is because it cannot find the training data. The training_config parameter should be the config file which specifies the data directory. To download the fma data, use the script download_fma_large.sh. There is also a small (7GB) FMA version which you can manually download instead.

fahnub commented 1 year ago

Thanks for this @MichaelALong

baardev commented 1 year ago

just create the folder data/fma_large (or whatever has been defined as "folder" in configs/training/train_musiclm_fma.json) and you will move past that error to the next error, "no files exists". Again, this mystery is solved by copying your 'mp3', 'wav', or 'flac' files to `data/fma_large'.

Suggestions to dev. instead of 'folder does not exist" and "no files exist", maybe say "folder 'data/fma_large' does not exist" and "No Wav, flac, MP3 files found in 'data/fma_large'"