查看来自Kaggle的数据和代码，似乎在训练和测试集的数据加载中都存在问题。首先，数据应该在每个标签的不同文件夹中，以便默认PyTorch
ImageFolder正确加载它。在您的情况下，由于所有训练数据都在同一文件夹中，因此PyTorch将其作为一个类加载，因此学习似乎很有效。您可以使用-
train/dog，-
train/cat，-
test/dog，-等文件夹结构来更正此问题
test/cat，然后将火车和测试文件夹
ImageFolder分别传递到火车和测试。训练代码看起来不错，只需更改文件夹结构，您就可以了。看一下ImageFolder的官方文档，其中有一个类似的示例。
