- 博客(2)
- 资源 (1)
- 收藏
- 关注
RSS订阅转载 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
3967
原创 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
10248
4
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人