본문 바로가기

include_top의 의미

The inception V3 has a fully-connected layer at the top. So by setting include_top to false, you're specifying that you want to ignore this and get straight to the convolutions.

->  top에 fully-connected layer가 있는 경우 include_top=False를 하면 무시하고 바로 convolution으로 가게 해 줌.

 

근데 fully-connected layer는 보통 bottom에 있지 top에 없지 않나? Tensorflow 공식문서에서 include_top 매개변수 설명을 찾아보니

 

whether to include the fully-connected layer at the top, as the last layer of the network

-> top에 fully-connected layer는 network의 가장 마지막 layer를 말한다.

 

아하, 그러면 여기서 말하는 top은 network의 bottom이랑 같은 개념이었다. 그렇다면 전이학습을 하기 위해 사전학습된 모델을 가져와 층을 더 추가하려면 include_top=False로 해주어야 한다.

참고


 

[1] DeepLearning.AI TensorFlow Developer Professional Certificate C2W3 Coding transfer learning from the inception mode

[2] Tensorflow 공식문서