add sparsity modules

This commit is contained in:
Victor SANH
2020-05-27 18:24:23 -04:00
parent 667003e447
commit e4c07faf0a
6 changed files with 1355 additions and 0 deletions

View File

@@ -0,0 +1,2 @@
from .binarizer import ThresholdBinarizer, TopKBinarizer, MagnitudeBinarizer
from .masked_nn import MaskedLinear