diff --git a/README.md b/README.md index c819dca7c1a70fbb1874154c7c7ea60612227912..b8bd6a3a49b1b66a981297399c5c4f37e8eb8756 100644 --- a/README.md +++ b/README.md @@ -60,8 +60,8 @@ Fine tunes the crepe model using the whole dataset. - [x] Loads 1024 sample windows and their corresponding f0 to be stored in a large `train_set.pkl` file (skip if data hasn't changed). - [x] Applies gradient descent using the BCE following the crepe paper (this task is treated as a binary classification for each spectral bin). - [x] The fine tuned model is stored in `crepe_ft/model_all.pth` -- [x] Train on one target species given as argument (weights are stored in `crepe_ft/model_only_{specie}.pth) -- [x] Train on all species except the target given as argument (weights are stored in `crepe_ft/model_omit_{specie}.pth) +- [x] Train on one target species given as argument (weights are stored in `crepe_ft/model_only_{specie}.pth`) +- [x] Train on all species except the target given as argument (weights are stored in `crepe_ft/model_omit_{specie}.pth`) ### Plotting Scripts allow to generate plots to visualise results (they are saved as `.pdf` files in the `figures` folder)