- 博客(3)
- 资源 (2)
- 收藏
- 关注
原创 tf.nn.in_top_k用法的一己之见
在处理一系列分类问题时,经常会用到tf.nn.in_top_k这一函数,其调用形式如下tf.nn.in_top_k(predictions, targets, k, name=None)其中:predictions: 你的预测结果(一般也就是你的网络输出值)大小是预测样本的数量乘以输出的维度target: 实际样本类别的标签,大小是样本数量的个数k: 每个样本...
2019-06-20 09:38:13 416
原创 使用gpu运算卷积网络时报错Failed to get convolution algorithm. This is probably because cuDNN....
在我们完成tensorflow-gpu配置后,我们希望使用其做一些实践,例如我希望可以使用它来加快之前跑起来极慢的facenet人脸识别实践。配置参数:cpu:e5v3gpu:2080ti内存:32G数据量:55万张当我使用--gpu_memory_fraction 0.9控制gpu内存使用率为90%时报错Failed to get convolution algorithm. Thi...
2019-06-12 09:44:29 6198 12
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人