问题描述
自己定义了一个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)
不太明白具体的原理。