mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
10 lines
178 B
Python
10 lines
178 B
Python
|
import random
|
||
|
|
import numpy as np
|
||
|
|
import torch
|
||
|
|
|
||
|
|
|
||
|
|
def random_seed(seed=42, rank=0):
|
||
|
|
torch.manual_seed(seed + rank)
|
||
|
|
np.random.seed(seed + rank)
|
||
|
|
random.seed(seed + rank)
|