feat: make max_grad_norm optional (None disables clipping)

- TrainConfig.max_grad_norm defaults to None - executor.clip_grad_norm returns grad norm without clipping when None - train.py --max_grad_norm defaults to None
2026-07-19 00:08:18 +08:00
parent fc62890e70
commit d7695b40e3
3 changed files with 21 additions and 6 deletions
@@ -148,8 +148,8 @@ def parse_args() -> argparse.Namespace:
    parser.add_argument(
        "--max_grad_norm",
        type=float,
-        default=1.0,
-        help="Max gradient norm for clipping.",
+        default=None,
+        help="Max gradient norm for clipping. None disables clipping.",
    )
    parser.add_argument(
        "--weight_decay",