Hello,
I am trying to get outputs from a GRU with:
input = embedded sequence of dimensions [146, 1, 256]
h_0 = hidden state of dimension [1, 1, 128]
I have worked with PyTorch GRUs before, but all of a sudden with these inputs I am getting the error cuDNN error: CUDNN_STATUS_EXECUTION_FAILED, even though the dimensions seem to match the requirements in the nn documentation, and all elements are cast to my ‘cuda’ device.
I have also checked my RAM usage and have plenty, so am confused.

Error.png1124×399 83.4 KB