Compatibility with PyTorch 1.5 and later

[madhawav]

Hi @henzler,

Have you tested training new models on PyTorch versions newer than 1.4?

I am doing a follow-up work based on this project and I have noticed that the model training performance degrades noticeably when I train in PyTorch 1.5 or later. Inference on models trained using PyTorch 1.4, via recent versions of PyTorch works just fine. I am guessing this issue is related to the noise operator implementation I obtained from this codebase.

[henzler]

Unfortunately, I have not tested if it works with a version newer than 1.4.