[Pytorch] Set seed

import torch
import numpy as np
import random
import torch.backends.cudnn as cudnn

def set_seed(num):
    torch.manual_seed(num)
    torch.cuda.manual_seed(num)
    torch.cuda.manual_seed_all(num)
    np.random.seed(num)
    cudnn.benchmark = False
    cudnn.deterministic = True
    random.seed(num)