그동안 직접 만든 CNN으로 계속 학습시키고, 결과를 노션에 혼자 정리했더니 블로그에 기록할 일이 많이 없었습니다. 그러던 중, CNN의 결과가 너무 안나와서 ResNet과 DenseNet을 전이학습해서 해보자는 조언을 얻어서, 이렇게 전이학습에 대해 포스팅을 하게 되었습니다! 기존 CNN 일단, 지금 내가 사용하고 있는 CNN은 다음과 같습니다 class ClassificationModel(nn.Module): '''CNN 직접 쌓아 만든 모델''' def __init__(self): super(ClassificationModel, self).__init__() self.feature_extraction = nn.Sequential( # [batch_size, 512,512,1] -> [batch_si..
