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