space

2020-05-27 20:44:45 -04:00
parent 5f8f2d849a
commit db2a3b2e01
2 changed files with 2 additions and 2 deletions
--- a/examples/movement-pruning/bertarize.py
+++ b/examples/movement-pruning/bertarize.py
@@ -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)",