今天本来是打算继续对CIFAR-10数据集进行操作,但是由于tensorflow的版本太高,再加上tf2将contrib这个库给删除了,导致训练代码不能运行
但是之前的测试代码,即将shuffle=True后即测试图片打乱顺序进行测试,如下图所示:
之后,为了不浪费这个数据集,就重新在github上找了一个pytorch的代码跑经过了一个半小时,200个epoch将准确率训练到92.39%
本次代码使用的是resnet训练,与书上的不同,但效果明显好得多,之后也会对这些进行理论方面的学习。