riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

space

Browse Source
This commit is contained in:
Victor SANH 2020-05-27 20:44:45 -04:00
parent 5f8f2d849a
commit db2a3b2e01
2 changed files with 2 additions and 2 deletions
Show Stats 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(
"--pruning_method",
choices=["l0", "magnitude", "topK", "sigmoied_threshold", ],
choices=["l0", "magnitude", "topK", "sigmoied_threshold",],
type=str,
required=True,
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(
"--pruning_method",
choices=["l0", "topK", "sigmoied_threshold", ],
choices=["l0", "topK", "sigmoied_threshold",],
type=str,
required=True,
help="Pruning Method (l0 = L0 regularization, topK = Movement pruning, sigmoied_threshold = Soft movement pruning)",