Update params.py: "BatchNorm" -> "LayerNorm" in the description strin…

…g for "--lock-text-freeze-layer-norm" (#767)
mlfoundations · Dec 21, 2023 · 695c72d · 695c72d
1 parent 73ad04a
commit 695c72d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/training/params.py b/src/training/params.py
@@ -386,7 +386,7 @@ def parse_args(args):
         "--lock-text-freeze-layer-norm",
         default=False,
         action='store_true',
-        help="Freeze BatchNorm running stats in text tower for any locked layers.",
+        help="Freeze LayerNorm running stats in text tower for any locked layers.",
     )
     parser.add_argument(
         "--log-every-n-steps",