Skip to content

Instantly share code, notes, and snippets.

@budui

budui/change_model_fc.py

Created February 22, 2019 06:56
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 budui/dc4e20e4e74f06ef46249d92f78f2ae5 to your computer and use it in GitHub Desktop.
Save budui/dc4e20e4e74f06ef46249d92f78f2ae5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
change_model_fc.py
# How to change a model's fc layer.
# with this code, you can get a model's features
del model.fc
model.fc = lambda x:x
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment