Configurable gradient clipping to avoid exploding gradients

Author: forklady42

src/electrai/entrypoints/train.py

@@ -93,6 +93,7 @@ def train(args):
         precision=cfg.model_precision,
         log_every_n_steps=10,
         profiler=profiler,
+        gradient_clip_val=getattr(cfg, "gradient_clip_value", 1.0),
     )
 
     # -----------------------------