Skip to content
Snippets Groups Projects
Commit c66ba838 authored by Paul Best's avatar Paul Best
Browse files

Update README.md

parent ac8aa2bc
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment