TensorFlow實現AlexNet卷積網絡模型

首先,我們先引入一些我們將會用到的庫,並定義-些會用到的變量。設置batch size爲32,表示每一輪訓練的batch中樣本的數目爲32。不像基於MNIST或Cifar-10數據集的實驗採用包含上百樣例的 batch,batch size設置爲32是因爲圖片數據比較大(高度和寬度都達到了224,深度達到了3),採用較大的 batch 會導致顯存不足的情況。num_batch是batch的數量,由
相關文章
相關標籤/搜索