Spaces:
Sleeping
Sleeping
🔨 [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,
|