Spaces:
Running on Zero
Running on Zero
File size: 173 Bytes
0d8b898 | 1 2 3 4 5 6 7 8 9 | import random
import torch
def fix_seed(seed: int):
random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed) |