🔨 [Update] gradient clip strategy
Browse files- yolo/lazy.py +1 -0
yolo/lazy.py
CHANGED
@@ -24,6 +24,7 @@ def main(cfg: Config):
|
|
24 |
logger=loggers,
|
25 |
log_every_n_steps=1,
|
26 |
gradient_clip_val=10,
|
|
|
27 |
deterministic=True,
|
28 |
enable_progress_bar=not getattr(cfg, "quite", False),
|
29 |
default_root_dir=save_path,
|
|
|
24 |
logger=loggers,
|
25 |
log_every_n_steps=1,
|
26 |
gradient_clip_val=10,
|
27 |
+
gradient_clip_algorithm="value",
|
28 |
deterministic=True,
|
29 |
enable_progress_bar=not getattr(cfg, "quite", False),
|
30 |
default_root_dir=save_path,
|