Would be good if the user could provide a string as model

[bischtob]

Users should be able to input either a WaveNet model, a string 'wavenet', 'gru', 'lstm', etc. No matter how cool WaveNet is, many users would expect to be able to use LSTM.