henry000 commited on
Commit
4b8ec68
·
1 Parent(s): 91a3f78

🌱 [Add] deterministic config for training

Browse files
Files changed (1) hide show
  1. 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: