在TensorFlow中使用CIFAR-10数据集训练VGG16,无法获得良好的准确性

时间:2018-11-26 07:00:34

标签: tensorflow

我是学习ML的初学者。我正在TensorFlow中研究VGG16模型。我的学习是here

现在我遇到一个问题,但仍然找不到原因。我研究了两种情况。

  1. 研究预训练的VGG16-A模型以进行预测。 (VGG in TensorFlow
  2. 使用CIFAR-10数据集研究VGG16-B模型以进行训练。 (Ashing00/Tensorflow_VGG

它们都工作正常。但是现在我要模拟Case 2的过程,尝试使用VGG16-A模型对CIFAR-10数据集进行训练。但是我失败了,我无法获得很好的准确性。

测试在vgg16f_training_cifar10.ipynb中进行。有人可以给我一些提示吗? 非常感谢。

1 个答案:

答案 0 :(得分:0)

我在每个隐藏层中添加批处理规范化。现在我们可以获得良好的准确性。