def set_seed(args):
    random.seed(args.seed)
    np.random.seed(args.seed)
    torch.manual_seed(args.seed)
    logger.info(f'Seed = {args.seed}')
    if (args.n_gpu > 0):
        torch.cuda.manual_seed_all(args.seed)
def set_seed(args):
    random.seed(args.seed)
    np.random.seed(args.seed)
    torch.manual_seed(args.seed)
    logger.info(f'Seed = {args.seed}')
    if (args.n_gpu > 0):
        torch.cuda.manual_seed_all(args.seed)