SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
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)",