movement-pruning/emmental/modules/__init__.py (
2
lines of code) (
raw
):
from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer from .masked_nn import MaskedLinear