diff --git a/src/taggingbackends/explorer.py b/src/taggingbackends/explorer.py index ff726b88eaafd022ab9225649c93fa1ac31bc5fd..6ce6cb49d6c91fcfcb78f0f46ae2d1ca188e70f4 100644 --- a/src/taggingbackends/explorer.py +++ b/src/taggingbackends/explorer.py @@ -213,6 +213,10 @@ Cannot find any Python package in project root directory: # typically follows: # UserWarning: CUDA initialization: CUDA unknown error print(line) + elif line == ' return F.conv2d(input, weight, bias, self.stride,': + # typically follows: + # UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED ... + print(line) # tensorflow logs elif 26 < len(line): # assume line[:26] to be e.g.: 2022-05-17 18:48:15.120981