
[Pytorch] 모델 작성과 모델 학습 과정 정리
·
ML_DL/딥러닝 공부하기
Seed 고정 seed = 22 random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False torch.backends.cudnn.deterministic = True - 같은 input에 대해 결과 재현이 가능하도록 한다. torch.backends.cudnn.benchmark = False - 하드웨어에 따라 최적화된 알고리즘을 사용할 수 있도록 한다. (따라서 옵션을 끈다) CustomDataset from torch.utils.dataset import..