A repository of the Music Transformer and Neural Transposer designed to be able to generate music in a relative major/minor depending on input.
- train_transformer.py: Located in /src/libs/.
- Arguments:
- numpy_path: A path to the .npy training files to be used. Default=../numpy_path
- save_path: A path where the model should be saved. Default=../models
- model_name: The name that the model should be saved under. Default={lr}{trainer}{epochs}
- epochs: The number of epochs that the model should be trained for. Default=10
- dset_dim: The subsection length to split dataset songs into. i.e. context window. Default=150
- d_model: The dimention of the model. Default=128
- num_layers: The number of layers within the Transformer. Default=2
- trainer: The name of the person who trained the model. Default=MusGen
- lr: The learning rate to be used in training. Default=0.6
- print_every: The epoch step number that when reached, prints training details out to console. Default=200
- Usage:
- python3 train_transformer.py --arg1 arg1val --arg2 arg2val ... etc.
- Arguments:
- /src/: finished/completed files, files belong here after believing that minimal changes will need to be made to them.
- /libs/: A place for helper files and functions. Additional definitions here as well, i.e. Transformer model.
- /mid_data_collections/: the collection of 10,000 .mid files separated into subsections for convenience.
- /misc/: A folder for random things.
- /models/: Where saved models are stored.
- /numpy_data_collections/: the collection of converted .midi files to .npy files for easier loading into training data, etc. Separated into subsections for convenience.
- /numpy_path/: The path where the complete set of converted .npy music files are stored to be read into program.
- /song_data/: Files are stored here that are related to the dataset, songnames, IDs, etc.
- /util/: A place for utility files to be stored. i.e., "create_npy.py" for .mid to .npy conversion.
- /dev/: A development folder used to develope files/methods/etc., store temp files, that when completed will be moved to '/src/'.