RNN state makes model output tend towards 0

[Tom2096]

Hi,

I have been following the examples in RNN-Denoise. However, when testing the quantized model, I noticed that the quantized output tends very fast towards zero:

Resetting the states of each RNN layer after each prediction makes the output more reasonable but does not give good noise reduction result:

Is there something I am missing? Note that the model used is the one provided in the repo and rest of the code is identical to the examples in RNN-Denoise. Thanks!