import torch import numpy as np import random def seed_all(seed=0): random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) np.random.seed(seed)