基于GoogLeNet v1的图像分类器
在开始训练之前请确保拥有足够的用于训练神经网络的图像,并将其分为训练集和测试集,在train_cpu或train_gpu中正确配置其路径
请在拥有独显的设备上采用train_gpu进行训练,这将成倍的提升训练效率,更快的得到分类模型。
使用gpu训练将会把所有代训练图像分批加载到内存中,请保证你拥有足够的内存空间,否则建议使用cpu版本进行训练,cpu版本采取从硬盘中加载的训练方式,当然,训练速度也会降低
请参照train_gpu中的代码注释进行正确的设置