TensorFlow实现AlexNet卷积网络模型

首先,我们先引入一些我们将会用到的库,并定义-些会用到的变量。设置batch size为32,表示每一轮训练的batch中样本的数目为32。不像基于MNIST或Cifar-10数据集的实验采用包含上百样例的 batch,batch size设置为32是因为图片数据比较大(高度和宽度都达到了224,深度达到了3),采用较大的 batch 会导致显存不足的情况。num_batch是batch的数量,由
相关文章
相关标签/搜索