tensorflow
文章平均质量分 85
随手笔记_000333
这个作者很懒,什么都没留下…
展开
-
tensorflow r1.5 版本差异调研
一、Breaking Changes 1、预编制的二进制文件是针对CUDA 9 和 cuDNN 7制定的。 2、Linux的二进制文件是使用 ubuntu 16构建的,在使用ubuntu 14引入glibc的时候,可能会出现不兼容的问题。 3、从1.6的版本开始,预编译的二进制文件会使用AVX指令,可能会破坏CPU上的TF。(This may break TF on older CPUs)翻译 2018-01-05 09:42:04 · 1799 阅读 · 0 评论 -
Tensor must be 4-D with last dim 1, 3, or 4,bug记录
torch学习入坑(1) 记录一下第一个坑,防止自己忘记。 在用torchvision.utils.make_grid处理完图片之后,维度会变为三维,如果需要使用tf.summary.image ,需要增加batch size 的那一维,需要用到的函数为 tf.expand_dims。 注意: 1、torchvision.utils.make_grid函数 输入:Tensor of s...原创 2018-04-16 18:55:56 · 2010 阅读 · 0 评论