reruff

BAMeScience · Jan 18, 2025 · 5c62d76 · 5c62d76
1 parent de5ef09
commit 5c62d76
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/equitrain/argparser.py b/equitrain/argparser.py
@@ -124,7 +124,10 @@ def add_optimizer_args(parser: argparse.ArgumentParser) -> argparse.ArgumentPars
         '--alpha', default=0.99, type=float, help='Smoothing constant (default: 0.99)'
     )
     parser.add_argument(
-        '--gamma', default=0.8, type=float, help='Multiplicative factor of learning rate decay (default: 0.8)'
+        '--gamma',
+        default=0.8,
+        type=float,
+        help='Multiplicative factor of learning rate decay (default: 0.8)',
     )
     parser.add_argument(
         '--momentum', default=0.9, type=float, help='SGD momentum (default: 0.9)'