diff --git a/utils/general.py b/utils/general.py index d4cd9a2..ec5717d 100755 --- a/utils/general.py +++ b/utils/general.py @@ -23,7 +23,8 @@ from scipy.cluster.vq import kmeans from scipy.signal import butter, filtfilt from tqdm import tqdm -from utils.torch_utils import init_seeds, is_parallel +from utils.torch_utils import init_seeds as init_torch_seeds +from utils.torch_utils import is_parallel # Set printoptions torch.set_printoptions(linewidth=320, precision=5, profile='long') @@ -55,7 +56,7 @@ def set_logging(rank=-1): def init_seeds(seed=0): random.seed(seed) np.random.seed(seed) - init_seeds(seed=seed) + init_torch_seeds(seed=seed) def get_latest_run(search_dir='./runs'):