Spaces:
Running
Running
🌱 [Add] deterministic config for training
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 |
)
|
| 28 |
|
| 29 |
match cfg.task.task:
|
|
|
|
| 24 |
logger=loggers,
|
| 25 |
log_every_n_steps=1,
|
| 26 |
gradient_clip_val=10,
|
| 27 |
+
deterministic=True,
|
| 28 |
)
|
| 29 |
|
| 30 |
match cfg.task.task:
|