Skip to content

Instantly share code, notes, and snippets.

@Sachin-crypto

Sachin-crypto/adding_output_layer.py

Created December 16, 2022 12:24
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Sachin-crypto/fb436c5587fcdda7351490a9bc944bfa to your computer and use it in GitHub Desktop.
Save Sachin-crypto/fb436c5587fcdda7351490a9bc944bfa to your computer and use it in GitHub Desktop.
Download ZIP
Raw
adding_output_layer.py
# Removing last 6 layers and adding our output layer
x = mobile.layers[-6].output
output = Dense(units=10, activation='softmax')(x)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment