From db2a3b2e01a91a5f37ad1922bc47ed4203375552 Mon Sep 17 00:00:00 2001 From: Victor SANH Date: Wed, 27 May 2020 20:44:45 -0400 Subject: [PATCH] space --- examples/movement-pruning/bertarize.py | 2 +- examples/movement-pruning/counts_parameters.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/movement-pruning/bertarize.py b/examples/movement-pruning/bertarize.py index 80643687b7..1fa99dcd95 100644 --- 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)", diff --git a/examples/movement-pruning/counts_parameters.py b/examples/movement-pruning/counts_parameters.py index 5b5dac9506..250850f352 100644 --- a/examples/movement-pruning/counts_parameters.py +++ b/examples/movement-pruning/counts_parameters.py @@ -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)",