19 lines
297 B
Python
19 lines
297 B
Python
import numpy as np
|
|
import torch
|
|
import random
|
|
|
|
|
|
def fixseed(seed):
|
|
random.seed(seed)
|
|
np.random.seed(seed)
|
|
torch.manual_seed(seed)
|
|
|
|
|
|
SEED = 10
|
|
EVALSEED = 0
|
|
# Provoc warning: not fully functionnal yet
|
|
# torch.set_deterministic(True)
|
|
torch.backends.cudnn.benchmark = False
|
|
|
|
fixseed(SEED)
|