ShuffleNet图像分类(2)
构建ShuffleNet网络
模型训练和评估
训练集准备与加载
采用CIFAR-10数据集对ShuffleNet进行预训练。
模型训练
本节用随机初始化的参数做预训练。首先调用定义网络,参数量选择,并定义损失函数为交叉熵损失,学习率经过4轮的后采用余弦退火,优化器采用。最后用中的接口将模型、损失函数、优化器封装在中,并用对网络进行训练。
模型评估
在CIFAR-10的测试集上对模型进行评估。
模型预测
在CIFAR-10的测试集上对模型进行预测,并将预测结果可视化。
学习时间与id: