标签: python tensorflow multi-gpu
我正在尝试为我的项目编写多GPU TensorFlow代码,并在TensorFlow回购中遇到了CIFAR10 code。我不太了解代码中tower_loss的用法。为什么我们不能只使用loss = cifar10.loss(logits, labels)代替第178行,并取所有损失的平均值?
tower_loss
loss = cifar10.loss(logits, labels)