자세한 설명은 다른 분들이 적어둔 걸 참고 하고
시간이 없는 나는 일단 적어두고 복붙
def seed_everything(seed=1):
""" Seed and GPU setting """
random.seed(seed)
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.deterministic = True
# torch.backends.cudnn.benchmark = True
'AI > pytorch' 카테고리의 다른 글
copy.deepcopy(model.state_dict()) (0) | 2022.09.28 |
---|