What's the purpose of include_top during Transfer Learning ? #111
ashikshafi08
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
To make it short
include_top = True
will remove the final dense layers (output layers) of the pre-trained model and we use the convolutional layers to extract features from our data and replace the dense layer with something of our own class.Read through this StackOverflow thread to get the detailed info about this.
Hope it helps.
Beta Was this translation helpful? Give feedback.
All reactions