transformers/examples/research_projects/bertology
Aleksey Tikhonov 291974c65c
GPT-model attention heads pruning example (#9189)
* Pruning for GPT attn heads

* The code formatted according to the transformers requirements

* Update run_prune_gpt.py

* Update run_prune_gpt.py
2020-12-18 16:32:10 -05:00
..
requirements.txt Reorganize examples (#9010) 2020-12-11 10:07:02 -05:00
run_bertology.py Reorganize examples (#9010) 2020-12-11 10:07:02 -05:00
run_prune_gpt.py GPT-model attention heads pruning example (#9189) 2020-12-18 16:32:10 -05:00