def set_seed()

in tbsm_pytorch.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def set_seed(seed, use_gpu):
    random.seed(seed)
    os.environ['PYTHONHASHSEED'] = str(seed)
    np.random.seed(seed)
    if use_gpu:
        torch.manual_seed(seed)
        torch.cuda.manual_seed(seed)
        torch.cuda.manual_seed_all(seed)   # if using multi-GPU.
        torch.backends.cudnn.benchmark = False
        torch.backends.cudnn.deterministic = True