데이터 준비하기 부분에서..
transform_train, transform_test만 있고,
transform_val은 선언되지 않았습니다.
또한 아래와 같이 code 수정 또는 보완이 된다면 좋을거 같습니다.
- 원본
train = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
test = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
val = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
- 수정필요
train = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
test = torchvision.datasets.ImageFolder(root='경로', transform=transform_test)
val = torchvision.datasets.ImageFolder(root='경로', transform=transform_val)
