首页 > 解决方案 > 迁移学习 Inception V3:如果准确性波动,是否继续训练?

问题描述

我仍在使用 tensorflow 上的 Inception V3 进行迁移学习。这是我使用 Tensorboard 进行训练的屏幕截图。
基础训练源码: https ://raw.githubusercontent.com/tensorflow/hub/master/examples/image_retraining/retrain.py

我的数据集约为 100 万,训练:验证:测试比率为 80:10:10。
32:128训练批量大小:验证批量大小
0.001 学习率

在此处输入图像描述

如您所见,训练准确度和验证准确度一直在波动(上下波动)。继续训练有什么意义吗?还是有什么导致我可以尝试解决的问题?

标签: tensorflowdeep-learningtraining-data

解决方案


看起来您的准确性正在发生很大变化。您使用的是哪种优化器?亚当?无论如何,如果您添加学习率衰减可能会有所帮助。


推荐阅读