![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
gorgeous___youth
这个作者很懒,什么都没留下…
展开
-
TensorFlow-抽取模型某一层特征 keras输出中间层结果的2种方法
TensorFlow-抽取模型某一层特征 原文链接深度学习具有强大的特征表达能力。有时候我们训练好分类模型,并不想用来进行分类,而是用来提取特征用于其他任务,比如相似图片计算。接下来讲下如何使用TensorFlow提取特征。1.必须在模型中命名好要提取的那一层,如下self.h_pool_flat = tf.reshape(self.h_pool, [-1, num_filter...转载 2018-10-29 10:22:48 · 22647 阅读 · 4 评论 -
理解tf.squeeze()
squeeze( input, axis=None, name=None, squeeze_dims=None)该函数返回一个张量,这个张量是将原始input中所有维度为1的那些维都删掉的结果axis可以用来指定要删掉的为1的维度,此处要注意指定的维度必须确保其是1,否则会报错>>>y = tf.squeeze(inputs, [0,...转载 2018-11-21 14:18:27 · 5875 阅读 · 0 评论 -
如何在基于tensorflow的深度学习框架keras中指定GPU
import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="0,1"原文转载 2018-11-26 15:47:13 · 190 阅读 · 0 评论 -
版本查看
查看cuda版本cat /usr/local/cuda/version.txt查看CUDNN版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看pytorch版本torch.__version__查看tensorflow版本tf.__version__Linux查看glibc安装版本...转载 2018-12-03 19:17:59 · 174 阅读 · 0 评论