It would be nice to have some augmentation option for small datasets: -random pitch shifter -random speed This would improve generalization of the model and have less chance of overfitting it. I'm not a coder tho, so i can't help you with that :D