Initial sync of refactor branch with master - Githubissues

orcasound / aifororcas-orcaml

Code for data preparation, training and evaluation of AI underlying Pod.Cast and OrcaHello projects.

MIT License

2 stars 3 forks source link

Initial sync of refactor branch with master #1

Closed akashmjn closed 4 years ago

akashmjn commented 4 years ago

cleanup download_datasets.py : easy download of latest train/test sets
major fixes in baseline Pytorch/AudioSet training code (code hasn't been touched much since July '19)
- track train/validation set losses
- logging with tensorboard
- bugfix for optimizer/scheduler -> big perf improvement
- clean progress bars
- support finetuning entire model
- averaging final model from multiple epochs
reorganize code & tweak README files a bit