Commit cc746a50 authored by Victor SANH's avatar Victor SANH
Browse files

flake8 compliance

parent b11386e1
...@@ -100,7 +100,7 @@ if __name__ == "__main__": ...@@ -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)",
......
...@@ -67,7 +67,7 @@ if __name__ == "__main__": ...@@ -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)",
......
# 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,
......
# flake8: noqa
from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer from .binarizer import MagnitudeBinarizer, ThresholdBinarizer, TopKBinarizer
from .masked_nn import MaskedLinear from .masked_nn import MaskedLinear
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment