/LayerMerge

Official PyTorch implementation of "LayerMerge: Neural Network Depth Compression through Layer Pruning and Merging" (ICML'24)

Primary LanguagePythonMIT LicenseMIT

Watchers