Question on the custom pruning function

[nooralahzadeh]

Hi,
Thanks for sharing your codebase.
I wonder what exactly is doing the following part of your code as "pruning_model_custom".

BERT-Tickets/squad_trans.py

Line 914 in 4d8e035

pruning_model_custom(model, mask)

Thanks for your help
Farhad