tensorflow GPU crashes for 0 batch size CUDNN_STATUS_BAD_PARAM
问题 This issue seem to be existing for a long time and lots of users are facing the issue. stream_executor/cuda/cuda_dnn.cc:444] could not convert BatchDescriptor {count: 0 feature_map_count: 64 spatial: 7 264 value_min: 0.000000 value_max: 0.000000 layout: BatchDepthYX} t o cudnn tensor descriptor: CUDNN_STATUS_BAD_PARAM The message is so mysterious that I do not know what happened in my code, however, my code works fine on CPU tensorflow. I heard that we can use tf.cond to get around this, but