- 博客(2)
- 收藏
- 关注
RSS订阅原创 Tensorflow -Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
我在做卷积层可视话的时候碰到的这个问题，其实不是cuDNN配置的问题，而是GPU占用，可通过下面的代码解决。 from tensorflow.compat.v1 import ConfigProto from tensorflow.compat.v1 import InteractiveSession config = ConfigProto() config.gpu_options.allow_growth = True session = InteractiveSession(config=co
2020-11-25 15:38:38
201
1
原创 Tensorflow——padding
tensorflow卷积中padding包括"valid"or"same". 关于这两种方式网上很多，这里不再赘述，贴上官方解释： "valid"means no padding."same"results in padding evenly to the left/right or up/down of the input such that output has the same height/width dimension as the input. 其实还有第三种方式：[[0,...
2020-07-17 19:14:33
120
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人