Modeling - [Error] 훈련 코드 오류 극복 과정 (이미지 사이즈, 배치 사이즈 조정)

[UGeunJi]

모델 층 간의 입출력 크기가 맞지 않아서 오류 발생

• Image_size 조절

trans = transforms.Compose([transforms.Resize((224, 224)),
                            transforms.ToTensor(),
                            transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))
                            ])

244 -> 32까지로 변경

• Batch_size 조절

batch_size = 16
trainloader = DataLoader(trainset, batch_size=batch_size, shuffle=True, num_workers=2) # 훈련용
validloader = DataLoader(valset, batch_size=batch_size, shuffle=False, num_workers=2) # 검증용
testloader = DataLoader(testset, batch_size=batch_size, shuffle=False, num_workers=2) # 테스트용

16 -> 2까지 변경

훈련 코드는 실행됐지만 32X32 이미지로는 훈련시킬 수가 없음.

결론적으로 전이학습을 결심하게 됨.