tensorflow
夕何
希望强人工智能早日到来,为此你我都需努力。
展开
-
如何使用特定显卡跑tensorflow
如何使用特定显卡跑tensorflow直入正题,有两种比较简单的方法方法一:改变系统变量使得仅目标显卡可见export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号,正常的话是从0开始然后再运行tensorflow即可方法二:使用tensorflow api添加设备环境with tensorflow.device('/gpu:0'): main tensorflo原创 2017-08-16 16:58:26 · 5397 阅读 · 0 评论 -
查看tensorflow ckpt文件中的变量名和对应值
查看tf ckpt文件中的变量名和对应值转载 2017-09-09 17:22:03 · 9469 阅读 · 0 评论 -
tensorflow 中 variable_scope 与name_scope函数解析
前言还是那句老话,学习tensorflow最好的方法就是阅读他的官方API手册。 知乎上面一个有意思的问答:tensorflow里面name_scope, variable_scope等如何理解?先引用知乎上答主的话: 主要是因为 变量共享 的需求。而这就不得不谈到tf. get_variable()了。因为如果使用Variable 的话每次都会新建变量,但是大多数时候我们是希望一些变量重用的原创 2017-09-18 22:27:31 · 2812 阅读 · 0 评论 -
使用tf.convert_to_tensor 时遇到value error 怎么解决
问题描述最近写tensorflow 程序,使用PIL读取图片后转存为numpy 数组,然后创建batch数据时使用的是list,即,list元素是array,然而在将这个list转换为tensor时,出现了这样的问题:....-got shape [64, 224, 224, 3], but wanted [64].函数描述convert_to_tensor( value, dtyp原创 2017-10-08 17:04:02 · 11886 阅读 · 1 评论