🔨 [Update] gradient clip strategy

WongKinYiu · Dec 23, 2024 · ee709f5 · ee709f5
1 parent c0e2436
commit ee709f5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/yolo/lazy.py b/yolo/lazy.py
@@ -24,6 +24,7 @@ def main(cfg: Config):
         logger=loggers,
         log_every_n_steps=1,
         gradient_clip_val=10,
+        gradient_clip_algorithm="value",
         deterministic=True,
         enable_progress_bar=not getattr(cfg, "quite", False),
         default_root_dir=save_path,