Skip to content

GitLab

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

space

parent 5f8f2d84
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/movement-pruning/bertarize.py
View file @ db2a3b2e
...@@ -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)",
......
examples/movement-pruning/counts_parameters.py
View file @ db2a3b2e
...@@ -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)",
......
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