search

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!
MIT License
360 stars 43 forks source link

Add Colab notebook example to analyze Huggingface model #33

Open noahgolmant opened 3 years ago

noahgolmant commented 3 years ago

Add a Colab notebook that measures the Hessian spectrum for a BERT or GPT-style language model.