파이썬 시드 고정
- EDA Study/코드
- 2020. 12. 25.
파이썬 및 파이토치의 시드를 고정해주는 명령어
# Update 코드
# 링크 : https://www.kaggle.com/reighns/complete-and-reusable-pytorch-pipeline
def seed_all(seed: int = 1930):
print("Using Seed Number {}".format(seed))
os.environ["PYTHONHASHSEED"] = str(
seed) # set PYTHONHASHSEED env var at fixed value
torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
torch.cuda.manual_seed(seed) # pytorch (both CPU and CUDA)
np.random.seed(seed) # for numpy pseudo-random generator
random.seed(
seed) # set fixed value for python built-in pseudo-random generator
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.enabled = False
def seed_worker(_worker_id):
worker_seed = torch.initial_seed() % 2**32
np.random.seed(worker_seed)
random.seed(worker_seed)
seed_all(seed=1994)
'EDA Study > 코드' 카테고리의 다른 글
2차원 리스트의 Transpose 구하기 (0) | 2020.12.29 |
---|---|
2차원의 배열을 1차원의 배열로 바꾸는 코드 (0) | 2020.12.28 |
구글 드라이브의 이미지 외부링크 추출 (0) | 2020.12.27 |
파이썬 메모리 줄이는 코드 (정형데이터) (0) | 2020.12.25 |
파이썬 시드 고정 (0) | 2020.12.25 |
파이썬 경고 무시 (0) | 2020.12.25 |