🌱 [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:
|