src/lmr/utils/seed.py

import numpy as np
import random
import torch

def set_seed(config):
    print('config.seed',config.seed)
    random.seed(config.seed)
    np.random.seed(config.seed)
    torch.manual_seed(config.seed)
    torch.cuda.manual_seed(config.seed)
    torch.cuda.manual_seed_all(config.seed)