def set_all_seed()

in picotron/utils.py [0:0]

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

def set_all_seed(seed):
    for module in [random, np.random]: module.seed(seed)
    torch.manual_seed(seed)
    if torch.cuda.is_available(): torch.cuda.manual_seed_all(seed)