![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
念珠响玉
这个作者很懒,什么都没留下…
展开
-
tensorflow 针对不同层设置不同学习率
var_list1 = [variables from first 5 layers] var_list2 = [the rest of variables] train_op1 = GradientDescentOptimizer(0.00001).minimize(loss, var_list=var_list1) train_op2 = GradientDescentOptimi...转载 2018-05-14 16:46:25 · 3987 阅读 · 0 评论 -
tensorflow 运行时内存持续增长(溢出)
tensorflow 训练部分代码几乎都采用循环,如果在循环中用到tf的op操作,就会导致节点越来越多,图的大小也不停的增大,故而内存会持续增长常见情况有几种1、sess.run(a+b) 写成sess.run(add_a_b)最好2、tf.train.Saver() 保存模型时候,不要直接写tf.train.Saver().save()/restore(),这样也会导致...原创 2018-05-08 14:45:12 · 10326 阅读 · 4 评论