This commit is contained in:
Victor SANH committed 2020-06-01 09:23:31 -04:00
1 parent 5f8f2d849a
commit db2a3b2e01
2 files changed
+2 -2

No files matched your search

+1 -1
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)",