Thank you this is very interesting.
I wonder about your initial train config, wouldn't it be more efficient with time in mind for demo purposes, to use more wav samples with less epochs?
We kept the epochs at 100 to demonstrate the negative consequence of overfitting training data, when doing the test or dev set evaluations. We could probably reduce that to ~50 though to save time in training :)