Skip to content

Commit

Permalink
update readme to reflect new preprocessing
Browse files Browse the repository at this point in the history
  • Loading branch information
zhvng authored Mar 26, 2023
1 parent e6faffd commit 72a2f44
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -94,13 +94,11 @@ To do this, fill in the `data_preprocessor_cfg` field in the config and set `use

```shell
python ./scripts/preprocess_data.py \
--stage all # stage(s) we want to preprocess for: all | semantic | coarse | fine
--model_config ./configs/model/musiclm_small.json \
--training_config ./configs/training/train_fma_preprocess.json \
--rvq_path PATH_TO_RVQ_CHECKPOINT \ # path to previously trained rvq
--kmeans_path PATH_TO_KMEANS_CHECKPOINT # path to previously trained kmeans
```
Note: make sure to process enough data for the number of training steps you're planning to run. Once the trainer runs out of data it cycles back to the beginning, but there is no random cropping in this case so the samples will be repeated.

## Inference
Generate multiple samples and use CLAP to select the best ones:
Expand Down

0 comments on commit 72a2f44

Please sign in to comment.