问题描述

自己定义了一个Dataset，检查自定义Dataset没有问题。训练集使用了自己定义的Dataset，但是测试集使用的是ImageFolder，会导致训练损失下降，但是验证损失不断上升。

# 获取数据集
train_data = MyData(train_set_path, transform=train_tf)
test_data = ImageFolder(test_set_path, transform=test_tf)

解决方法

获取数据集时使用相同的获取方式

train_data = MyData(train_set_path, transform=train_tf)
test_data = MyData(test_set_path, transform=test_tf)

train_data = ImageFolder(train_set_path, transform=train_tf)
test_data = ImageFolder(test_set_path, transform=test_tf)

不太明白具体的原理。