def set_seed()

in run_generation.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

def set_seed(args):
    np.random.seed(args.seed)
    torch.manual_seed(args.seed)
    if args.n_gpu > 0:
        torch.cuda.manual_seed_all(args.seed)