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

flake8 compliance

Browse Source
This commit is contained in:
Victor SANH
2020-05-28 00:37:09 -04:00
parent b11386e158
commit cc746a5020
4 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/movement-pruning/bertarize.py
View File

@@ -100,7 +100,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"--pruning_method", "--pruning_method",
choices=["l0", "magnitude", "topK", "sigmoied_threshold",], choices=["l0", "magnitude", "topK", "sigmoied_threshold"],
type=str, type=str,
required=True, required=True,
help="Pruning Method (l0 = L0 regularization, magnitude = Magnitude pruning, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)", help="Pruning Method (l0 = L0 regularization, magnitude = Magnitude pruning, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)",

2
examples/movement-pruning/counts_parameters.py
View File

@@ -67,7 +67,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"--pruning_method", "--pruning_method",
choices=["l0", "topK", "sigmoied_threshold",], choices=["l0", "topK", "sigmoied_threshold"],
type=str, type=str,
required=True, required=True,
help="Pruning Method (l0 = L0 regularization, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)", help="Pruning Method (l0 = L0 regularization, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)",

1
examples/movement-pruning/emmental/__init__.py
View File

@@ -1,3 +1,4 @@
# flake8: noqa
from .configuration_bert_masked import MaskedBertConfig from .configuration_bert_masked import MaskedBertConfig
from .modeling_bert_masked import ( from .modeling_bert_masked import (
MaskedBertForMultipleChoice, MaskedBertForMultipleChoice,

1
examples/movement-pruning/emmental/modules/__init__.py
View File

@@ -1,2 +1,3 @@
# flake8: noqa
from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer
from .masked_nn import MaskedLinear from .masked_nn import MaskedLinear