Skip to content

Instantly share code, notes, and snippets.

@n0obcoder

n0obcoder/understanding_pytorch___04___freeze_all_but_fc.py

Last active August 13, 2019 06:51
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save n0obcoder/9b70c73065734d4ee189ea15e82a3fbd to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save n0obcoder/9b70c73065734d4ee189ea15e82a3fbd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
understanding_pytorch___04___freeze_all_but_fc.py
for name, param in model.named_parameters():
if name in ['fc.weight', 'fc.bias']:
param.requires_grad = True
else:
param.requires_grad = False
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment