![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
睡觉不准打呼噜
小白养成日记
展开
-
tensorflow 张量插入或者删除的办法
如下所示part1 = new_enc_in[:, 0:3 ]part2 = new_enc_in[:, 3:]###part_new 要插入的张量new_enc_in = tf.concat([part1, part_new, part2], 1)##得到的张量里part_new为新插入列##如果要删除,则修改part2的范围part1 = new_enc_in[:, 0...原创 2018-11-20 15:19:57 · 11545 阅读 · 0 评论 -
tensorflow 反向传播求导
X=tf.constant([-1,-2],dtype=tf.float32)w=tf.Variable([2.,3.])truth=[3.,3.]Y=w*X# cost=tf.reduce_sum(tf.reduce_sum(Y*truth)/(tf.sqrt(tf.reduce_sum(tf.square(Y)))*tf.sqrt(tf.reduce_sum(tf.square(tru...原创 2018-03-20 17:48:21 · 1475 阅读 · 0 评论 -
tensorflow 训练时内存溢出问题
我最近在尝试写FCN时,用了两层5*5的卷积和两层5*5的反卷积,训练数据是2000张图片,每次循环输入一张图片,输出一个数字表示本次循环结束,发现程序越跑越慢,原因是内存持续增大了。网上搜了一下发现,没几个是对的,于是只能自己填坑(有个sess.graph.finalize()的,没用)。于是debug,发现每次循环输入图片后,每张图片前向传播产生的中间map都会保存下来,不会被清理掉(就是说每原创 2018-01-18 10:24:42 · 13590 阅读 · 0 评论 -
对抗训练、领域适应中梯度反转的实现与测试
1.梯度反转的目标是:(1)正向传播时传递权值不变(2)反向传播时,神经元权值增量符号取反,即与目标函数方向切好相反达到对抗的目的2.梯度反转的实现:import tensorflow as tffrom tensorflow.python.framework import opsclass FlipGradientBuilder(object): def __i...原创 2019-06-11 14:14:02 · 3156 阅读 · 2 评论