Source code for torch_geometric.seed

import random

import numpy as np
import torch


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