How to prune a model trained with LayerDrop
I am training a BART transformers with LayerDrop for encoder and decoder.Now, I want to know how can I do the pruning? I did not find enough information in the documentation.