henry000 commited on
Commit
598b827
·
1 Parent(s): e530629

🔧 [Add] const random seed

Browse files
Files changed (1) hide show
  1. yolo/utils/logging_utils.py +1 -0
yolo/utils/logging_utils.py CHANGED
@@ -67,6 +67,7 @@ def set_seed(seed):
67
 
68
  class ProgressLogger(Progress):
69
  def __init__(self, cfg: Config, exp_name: str, *args, **kwargs):
 
70
  local_rank = int(os.getenv("LOCAL_RANK", "0"))
71
  self.quite_mode = local_rank or getattr(cfg, "quite", False)
72
  custom_logger(self.quite_mode)
 
67
 
68
  class ProgressLogger(Progress):
69
  def __init__(self, cfg: Config, exp_name: str, *args, **kwargs):
70
+ set_seed(cfg.lucky_number)
71
  local_rank = int(os.getenv("LOCAL_RANK", "0"))
72
  self.quite_mode = local_rank or getattr(cfg, "quite", False)
73
  custom_logger(self.quite_mode)