examples/research_projects/bertology/run_prune_gpt.py · 291974c65cb9d93587826f80ed7d401f2bab2de7 · chenpangpang / transformers

Aleksey Tikhonov authored Dec 18, 2020

* Pruning for GPT attn heads

* The code formatted according to the transformers requirements

* Update run_prune_gpt.py

* Update run_prune_gpt.py

291974c6

run_prune_gpt.py 15.5 KB