Source code for torch_geometric.seed

import random

import torch
import numpy as np


[docs]def seed_everything(seed: int): r"""Sets the seed for generating random numbers in PyTorch, numpy and Python. Args: seed (int): The desired seed. """ random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed_all(seed)