nelaturuharsha/TurboPrune

Harness for training/finding lottery tickets in PyTorch. With support for multiple pruning techniques and augmented by distributed training, FFCV and AMP.

语言：Python · GitHub 星标：19 · Fork：1

话题：distributed · lottery-ticket-hypothesis · pruning · pytorch

在 GitHub 上查看原仓库