File size: 526 Bytes
6f2c823
9589cd1
e4c0874
 
 
 
3356688
9589cd1
 
 
 
 
 
 
 
 
 
e4c0874
9589cd1
14ae0ea
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

## 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/)
`./download_egfx.sh`

## Train model
1. Change Wandb variables in `shell_vars.sh`
2. `python train.py exp=audio_diffusion`
or
2. `python 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`