15_코드 실행 시 오류 해결 방법

<강의 코드>

transforms_for_train = transforms.Compose(
        [
            transforms.Resize(feature_extractor.size),
            transforms.RandomHorizontalFlip(),
            transforms.ToTensor(),
            normalize,
        ]
    )

강의 코드를 변경하지 않고 그대로 실행하면 아래와 같은 오류가 발생합니다

<오류 메시지>

TypeError: Size should be int or sequence. Got

<해결 방법>

feature_extractor.size를 tuple(feature_extractor.size.values()) 로 변경한 뒤 실행합니다.

• transforms.Resize 내의 인자 뿐만 아니라 Crop 안의 인자도 함께 변경해주어야 합니다

   

transforms_for_train, transforms_for_val 모두 동일하게 변경한 뒤 실행하면 에러 없이 실행되는 것을 보실 수 있습니다.