layerInput = tf.nn.conv2d(model_inputs, # [batch,1,length,100] filter_weights, strides=[1, 1, 1, 1], padding="SAME", name="init_layer", use_cudnn_on_gpu=True)
在tensorflow2.0中此代码报错,把use_cudnn_on_gpu=True删除就可以了
layerInput = tf.nn.conv2d(model_inputs, # [batch,1,length,100] filter_weights, strides=[1, 1, 1, 1], padding="SAME", name="init_layer", use_cudnn_on_gpu=True)
在tensorflow2.0中此代码报错,把use_cudnn_on_gpu=True删除就可以了