🔧 [Add] const random seed
Browse files
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)
|