🌱 [Add] a random seed for training

yolo/utils/logging_utils.py

@@ -12,12 +12,14 @@ Example:
 """
 
 import os
+import random
 import sys
 from collections import deque
 from pathlib import Path
 from typing import Any, Dict, List
 
 import numpy as np
+import torch
 import wandb
 import wandb.errors.term
 from loguru import logger
@@ -49,6 +51,18 @@ def custom_logger(quite: bool = False):
     )
 
 
+# TODO: should be moved to correct position
+def set_seed(seed):
+    random.seed(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    if torch.cuda.is_available():
+        torch.cuda.manual_seed(seed)
+        torch.cuda.manual_seed_all(seed)  # if you are using multi-GPU.
+    torch.backends.cudnn.deterministic = True
+    torch.backends.cudnn.benchmark = False
+
+
 class ProgressLogger(Progress):
     def __init__(self, cfg: Config, exp_name: str, *args, **kwargs):
         local_rank = int(os.getenv("LOCAL_RANK", "0"))