[Colab x PyTorch] Music Transformer で AIに作曲・演奏してもらおう！ [Part02]

前回に引き続き、Music Transformerを使ってAIに作曲・演奏してもらう方法について紹介します。

手順2: モデルの学習と生成

まずは、Colab上でPyTorchやMusic Transformerのライブラリをインストールします。

!pip install torch torchvision torchaudio
!pip install transformers

次に、学習させるためのデータセットを用意しましょう。 MIDIファイルのデータセットをダウンロードして、Music Transformerが学習できる形式に変換します。

!wget http://www.iro.umontreal.ca/~lcharlin/datasets/nottingham_database.zip
!unzip nottingham_database.zip
!python preprocess_midi.py --input_dir=data/ --output_dir=dataset_np --config=default

最後に、モデルを学習させます。 ColabのGPUを使って学習させることができます。

!python train.py --config=default

学習が終わったら、モデルを使って新しい楽曲を生成することができます。

!python generate.py --config=default --checkpoint=best.pth.tar

まとめ

Music Transformerを使ってAIに作曲・演奏してもらう方法について、手順2を紹介しました。次回は、AIが生成した楽曲を聴いてみましょう！