[PyTorch] MNIST with Dropout
[Dropout] (class) torch.nn.Dropout(p=0.5, inplace=False) Overfitting을 해결할 수 있는 방법 중 하나. Ensemble의 효과를 낼 수 있음. (매 학습마다 형태가 변형 되기 때문에) Parameters p – probability of an element to be zeroed. Default: 0.5 inplace – If set to True, will do this operation in-place. Default: False !!!!중요!!!! train 시작 전 model.train()으로 dropout 사용을 명시해줘야 함. 또한 test 전 model.eval()로 dropout 사용안함을 명시해줘야 함. import torch impo..
