반응형
Pytorch 학습 시, seed를 고정하는 방법은 아래와 같다.
로스가 항상 같이 나오니 비교하기 좋다.
import random
SEED = 10
random.seed(SEED)
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)
torch.backends.cudnn.deterministic = True
반응형
'프로그래밍 > 기타' 카테고리의 다른 글
| [오류 해결] 'SSLError(SSLCertVerificationError(1, '[SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed (0) | 2022.06.13 |
|---|---|
| [Pytorch] Variable 에 관하여 (0) | 2022.06.08 |
| [오류 해결] OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized. (0) | 2022.06.08 |
| [pytorch] 개발환경 (0) | 2022.05.04 |
| [Dbeaver] CSV 데이터 가져오기 한글 깨짐 현상 방지 (0) | 2022.04.01 |