Skip to content

Instantly share code, notes, and snippets.

@Paulescu

Paulescu/load_model.py

Created June 24, 2022 13:25
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 Paulescu/e6e7e3619868cadb66894f7883126817 to your computer and use it in GitHub Desktop.
Save Paulescu/e6e7e3619868cadb66894f7883126817 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_model.py
from transformers import PerceiverFeatureExtractor, PerceiverForImageClassificationLearned
feature_extractor = PerceiverFeatureExtractor.from_pretrained("deepmind/vision-perceiver-learned")
model = PerceiverForImageClassificationLearned.from_pretrained("deepmind/vision-perceiver-learned")
n_params = sum(p.numel() for p in model.parameters())
print(f'Number of parameters: {n_params:,}')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment