首页 > 其他 > 详细

TF Multi-GPU single input queue

时间:2019-04-24 20:18:28      阅读:227      评论:0      收藏:0      [点我收藏+]

多GPU的数据训练,feed

 

images, labels = cifar10.distorted_inputs()
split_images = tf.split(images, FLAGS.num_gpus, 0)
split_labels = tf.split(labels, FLAGS.num_gpus, 0)

 

for i in xrange(FLAGS.num_gpus):
  with tf.device(/gpu:%d % i):
    with tf.name_scope(%s_%d % (cifar10.TOWER_NAME, i)) as scope:

      loss = tower_loss(scope, split_images[i], split_labels[i])

 

https://stackoverflow.com/questions/34273951/tensorflow-multi-gpu-single-input-queue?

--

 

TF Multi-GPU single input queue

原文:https://www.cnblogs.com/xiaoniu-666/p/10764493.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!