当前位置：网站首页>TypeError: Got an unexpected keyword argument ‘use_cudnn_on_gpu‘

TypeError: Got an unexpected keyword argument ‘use_cudnn_on_gpu‘

2022-08-06 20:50:00 【淋雨的小蚂蚁】

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删除就可以了

原网站

版权声明
本文为[淋雨的小蚂蚁]所创，转载请带上原文链接，感谢
https://blog.csdn.net/fighting_qq/article/details/126142878