（小白）用VGG16训练自己的数据集遇到了问题

怎么下载VGG代码、创建模型主要参考了这篇博客https://blog.****.net/weixin_42795611/article/details/91126616
数据集是我自己用labelimg创建的，一共9类

但是在运行VGG16.py出现如下问题：

在网上搜了一圈也没找到解决办法，后来把代码重头看一遍，原来是下面最后这行代码出了问题，因为自己的训练种类9类，要把softmax前面对应的输出也写成9，否则就会出现像上面这样维度不匹配的问题