Pytorch實現GoogLeNet解決各類數據集(cifar10/mnist/imagenet)分類

有一段時間沒有更新了，這次我給大家帶來的是大名鼎鼎的GoogleNet模型。也可以稱爲Inception v3模型。參考了源代碼，寫出了可讀性與性能更優的模型，在模型上有些許微調，輸入的圖片大小是224 x 224 x 3.!!! 圖片大小可以任意調節，甚至用來訓練CIFAR10也可以。本文先放出代碼。 完整代碼點這裏 ) import torch from torch import nn NU

>>阅读原文<<