rnnoise icon indicating copy to clipboard operation
rnnoise copied to clipboard

modify rnn_train.py script, vad_gru and denoise_gru will use relu as …

Open pyu1538 opened this issue 5 years ago • 2 comments

…activation, so it matches the paper

pyu1538 avatar Jun 05 '20 01:06 pyu1538

Doing so will prevent the kernel from using cuDNN version of GRU, defaulting to the generic implementation. That in turn will severely slow down the training (assuming you're using GPU for the training).

tracek avatar Jun 07 '20 12:06 tracek

Doing so will prevent the kernel from using cuDNN version of GRU, defaulting to the generic implementation. That in turn will severely slow down the training (assuming you're using GPU for the training).

Got it. I trained it without GPU, so I didn't feel slow down. RNNoise demo is also trained by "relu", it's better to use exact same model for comparison.

pyu1538 avatar Jun 10 '20 14:06 pyu1538