README.md

Install Packages

1. python3 -m venv env
2. source env/bin/activate
3. pip install -e .
4. pip install -e umx

Download [GuitarFX Dataset] (https://zenodo.org/record/7044411/)

./scripts/download_egfx.sh

Train model

1. Change Wandb variables in shell_vars.sh
2. python scripts/train.py exp=audio_diffusion or
3. python scripts/train.py exp=umx

To add gpu, add trainer.accelerator='gpu' trainer.devices=-1 to the command-line

Ex. python train.py exp=umx trainer.accelerator='gpu' trainer.devices=-1