F tensorflow/stream_executor/cuda/cuda_dnn.cc:522] Ошибка проверки
Я тренирую свою модель в тензорном потоке. После некоторых итераций я получаю следующее исключение:
F tensorflow/stream_executor/cuda/cuda_dnn.cc:522] Ошибка проверки: cudnnSetTensorNdDescriptor(handle_.get(), elem_type, nd, dims.data(), strides.data()) == CUDNN_STATUS_SUCCESS (3 против 0)batch_descriptor : {количество: 1 feature_map_count: 1088 пространственные: 63 0 value_min: 0,000000 value_max: 0,000000 layout: BatchDepthYX}
Проблема не в библиотеке cuda, так как тот же код работает и для других наборов данных. Что может быть причиной этого?
Были ли размеры слишком малы? Спасибо
привет Pervaiz, я столкнулся с той же проблемой. Не могли бы вы поделиться, как вы это исправляете? Благодарность
Проверьте размер ввода, если он меньше фактического ввода, попробуйте увеличить размер ввода, чтобы избежать этой ошибки. Спасибо
Проблема заключалась в размерах изображения после изменения размера. Это может кому-то помочь.