SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

commplying with isort

Browse Source
This commit is contained in:
Victor SANH
2020-05-28 00:26:39 -04:00
parent db2a3b2e01
commit 5c8e5b3709
9 changed files with 29 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/movement-pruning/emmental/modules/masked_nn.py
View File

@@ -19,14 +19,14 @@ the weight matrix to prune a portion of the weights.
The pruned weight matrix is then multiplied against the inputs (and if necessary, the bias is added).
"""
import math
import torch
from torch import nn
from torch.nn import functional as F
from torch.nn import init
import math
from .binarizer import ThresholdBinarizer, TopKBinarizer, MagnitudeBinarizer
from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer
class MaskedLinear(nn.Linear):