dataset/reseed.py

import torch
import random

def reseed(seed):
    random.seed(seed)
    torch.manual_seed(seed)