search

NVIDIA / nv-wavenet

Reference implementation of real-time autoregressive wavenet inference
BSD 3-Clause "New" or "Revised" License
735 stars 126 forks source link

Is it possible to do re-training or fine-tune in this code ? #47

Closed rishikksh20 closed 6 years ago

rishikksh20 commented 6 years ago

I want to do transfer learning, is it possible to do using its code? If yes then how nothing mentioned in README about re-training or fine-tune.

gsoul commented 6 years ago

What task would you like to "transfer-learn" this model for?

rishikksh20 commented 6 years ago

@gsoul I want to train this model on Indian voice, but I don't want to train the model from scratch. Instead, I want to re-train this model using pre-trained weights on Indian English Voice dataset.

gsoul commented 6 years ago

Well, theoretically you could do that: https://github.com/NVIDIA/nv-wavenet/blob/master/pytorch/config.json#L9

but given that you would need to retrain not only WaveNet, but also Tacotron model, I have some doubts on how well would that work. But of course, you never know until you try it.