Export to onnx

[bzp83]

Hi!

Thanks for this work!

I'm using this in a model and when I try to export the model to onnx, I get:
Missing key(s) in state_dict: "model_g.dec.stft.forward_basis", "model_g.dec.stft.inverse_basis".

If I change the lines below to persistent=False, the export works... But I'm not sure if this will have any effect that I can't foresee as this persistent flag is not clear to me.

Would you please tell me if this is acceptable?

Thank you!

torch-stft/torch_stft/stft.py

Line 53 in 193ad6a

self.register_buffer('forward_basis', forward_basis.float())

and

torch-stft/torch_stft/stft.py

Line 54 in 193ad6a

self.register_buffer('inverse_basis', inverse_basis.float())

to:

        self.register_buffer('forward_basis', forward_basis.float(), persistent=False)
        self.register_buffer('inverse_basis', inverse_basis.float(), persistent=False)