tensorflow
xz1308579340
这个作者很懒,什么都没留下…
展开
-
tf.contrib.training.train的使用
文章目录1.作用2.使用train的简单的例子2.1载入数据和创建模型2.2定义loss2.3定义优化器2.4创建train_op2.5运行训练过程3.如何定义train_op3.1创建一个train_op,对梯度进行裁剪操作3.2创建train_op并通过提供来自变量的映射来缩放梯度4训练过程进行非梯度更新5.从指定的checkpoint恢复模型6.从变量名不匹配的检查点初始化模型7.从chec...原创 2019-05-15 12:08:34 · 3786 阅读 · 6 评论 -
tensorflow中使用指定的GPU及GPU显存
tensorflow中使用指定的GPU及GPU显存本文目录1 终端执行程序时设置使用的GPU2 python代码中设置使用的GPU3 设置tensorflow使用的显存大小3.1 定量设置显存3.2 按需设置显存转载请注明出处:http://www.cnblogs.com/darkknightzh/p/6591923.html转载 2017-07-28 19:12:11 · 3160 阅读 · 0 评论