xyz961014 / PrunedTransformer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Prune Before Converged

In this work, we propose a structured pruning method to train Transformer models for neural machine translation efficiently. We construct experiments on two widely-used machine translation datasets, and results show that pruning early before convergence significantly saves total training time while keeping comparable performance.

About


Languages

Language:Python 93.8%Language:Shell 6.1%Language:Dockerfile 0.1%