机器学习
YaoEmily
脑洞少女 + 程序媛
展开
-
iteration, batchsize, epoch定义、区别和联系
iteration: 数据进行一次前向-后向的训练,就是更新一次参数; batchsize:每次迭代(iteration)训练图片的数量; epoch:1个epoch就是将所有的训练图像全部通过网络训练一次。例:有1280000张图片,batchsize=256,则1个epoch需要1280000/256=5000次iteration,假如它的max_iteration=450000,则共...原创 2018-05-26 13:28:50 · 821 阅读 · 1 评论 -
Tensorflow指定GPU训练
以下假设使用第3块GPU训练 1.在代码中添加 第一种方法: tf.device(‘/gpu:2’) 第二种方法: import os os.environ[‘CUDA_VISIBLE_DEVICES’]=’2’ 如果要指定1,3块GPU来训练,则: os.environ[‘CUDA_VISIBLE_DEVICES’]=’0,2’2.在命令中...原创 2018-05-24 09:57:03 · 2948 阅读 · 1 评论