random seed

heyme·2023년 1월 6일
0

딥러닝

목록 보기
10/10

def seed(self, seed_value):

import random
import torch.backends.cudnn as cudnn

random.seed(seed_value)
torch.manual_seed(seed_value)
torch.cuda.manual_seed_all(seed_value)
torch.cuda.manual_seed(seed_value)
np.random.seed(seed_value)
torch.backends.cudnn.deterministic = True
torch.backends.benchmark = False

0개의 댓글