Rate this Page
★ ★ ★ ★ ★

apply_fake_sparsity#

torchao.sparsity.apply_fake_sparsity(model, **kwargs)[source][source]#

This function simulates 2:4 sparsity on all linear layers in a model.

On this page
Edit on GitHub
Show Source