Loss.backword() shows RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

Hello there,
I’m getting RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED in loss.backward() stage, which has been triggered from autograd module of torch. What’s the reason behind it?

The executed script is: ZeroSpeech/train.py at 0ef25e4daac07888d22b802c649480dc70b719fe · bshall/ZeroSpeech · GitHub

My specs are:
GPU 1660s
CUDA 10.2
cudnn 7.6.5

Thanks in advance.

Could you create a minimal executable code snippet, which would reproduce the issue?
I.e. could you post the model configs or how the modules are initialized as well as the used input shapes?

Here is code snippet I’m using and this is the input part of the model:

The argument values which have been passed to this object is:
in_channels: 80
channels: 768
n_embeddings: 512
embedding_dim: 64
jitter: 0.5

The models are initialised here:

Let me know if anything you need to assure.

By the way, this is a appreciation for you @ptrblck. Whenever I face any pytorch issues and look for the solution, it’s often you who comes with the best solution. Thanks for everything bud!