CUDNN_STATUS_EXECUTION_FAILED while using convolution2dLayer with 2D image sequence(SSCBT)

Question

0 个投票

lgraph = [...
    sequenceInputLayer([64 5 1], "MinLength",4)
    % convolution2dLayer([3,3],1) % If include this conv2d layer cause
    % ERROR CUDNN_STATUS_EXECUTION_FAILED
    flattenLayer
    gruLayer(16)
    fullyConnectedLayer(16)
    fullyConnectedLayer(2)
    softmaxLayer
    classificationLayer
    ]

If include `convolution2dLayer` will throw ERROR:

Unexpected error calling cuDNN: CUDNN_STATUS_EXECUTION_FAILED

lgraph = [...
    sequenceInputLayer([64 5 1], "MinLength",4)
    sequenceFoldingLayer('Name','fold')
    convolution2dLayer([3,3],1)
    sequenceUnfoldingLayer('Name','unfold')
    flattenLayer
    gruLayer(16)
    fullyConnectedLayer(16)
    fullyConnectedLayer(2)
    softmaxLayer
    classificationLayer
    ]
lgraph = connectLayers(lgraph,'fold/miniBatchSize','unfold/miniBatchSize');