我正在尝试实现一个残差网络来对项目的 CIFAR10 数据集上的图像进行分类,并且我有一个工作模型,该模型的准确度呈对数增长,但验证准确度处于稳定状态。我在大多数层之后使用了批量标准化和 relu,最后使用了 softmax。
这是我的数据拆分:
(train_images, train_labels), (test_images, test_labels) = datasets.cifar10.load_data()
这是我编译和训练模型的代码
resNet50.compile(optimizer='adam',loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),metrics=['accuracy'])
resNet50.fit(train_images, train_labels, epochs=EPOCHS, validation_data=(test_images, test_labels))
什么可能导致这个验证平台,什么可以改进我的模型?
预先感谢您的反馈和评论。