search

VIPL-Audio-Visual-Speech-Understanding / learn-an-effective-lip-reading-model-without-pains

The PyTorch Code and Model In "Learn an Effective Lip Reading Model without Pains", (https://arxiv.org/abs/2011.07557), which reaches the state-of-art performance in LRW-1000 dataset.
152 stars 37 forks source link

CUDNN_STATUS_BAD_PARAM error #16

Closed marziehoghbaie closed 3 years ago

marziehoghbaie commented 3 years ago

HI there, I can only run the test function on cpu device. Have some one faced this error. even when I train the model, it works fine in train mode but when it calls the test function on validation set, it return 'CUDNN_STATUS_BAD_PARAM ' error. I searched and one solution was that to .float() to all your tensors. I did try this solution, but I'm still seeing the same error. Do any one have any idea?

marziehoghbaie commented 3 years ago

Hi again, I found the solution; I replaced with autocast(): in test function with with torch.no_grad(): and now it works fine on GPU.