강의

멘토링

커뮤니티

인프런 커뮤니티 질문&답변

meshjo님의 프로필 이미지
meshjo

작성한 질문수

딥러닝 CNN 완벽 가이드 - TFKeras 버전

Pretrained 모델의 Fine-tuning을 통한 모델 최적화

Xception 같은 경우 동결은 어떻게 해야될까요?

작성

·

362

0

Xception 모델 같은 경우 어디 layer 까지 trainable를 False로 지정해야 될까요?

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까,

보통 fine tuning의 첫번째 freeze는 CNN Layer를 합니다. 즉 Dense layer 구간은 먼저 학습하고, 다음에 CNN + Dense 포함한 모든 layer를 다 학습하는 방식이 fine-tuning 방식입니다.

강의와 비슷하게 Xception의 경우도 GlobalAveragePooling2D -> Dense 이전의 Layer는 첫번째 freeze로 포함시켜 주시면 됩니다. 그러니까 Dense layer나오기 전의 GlobalAveragePooling2D 이전의 Layer들을 다 freeze 해주시면 될 것 같습니다.

 

감사합니다.

meshjo님의 프로필 이미지
meshjo

작성한 질문수

질문하기