딥러닝 활성화함수(Relu , Maxpool2d)
class CNN(nn.Module): def __init__(self): super(CNN, self).__init__() self.layer1 = torch.nn.Sequential( torch.nn.Conv2d(1, 8, kernel_size=3,stride=1,padding=1), torch.nn.ReLU(), torch.nn.MaxPool2d(2, 2) ) self.layer2 = torch.nn.Sequential( torch.nn.Conv2d(8, 16, kernel_size=3, stride=1, padding=1), torch.nn.ReLU(), torch.nn.MaxPool2d(2, 2) ) self.fc = torch.nn.Sequential( torch.nn.Linear(40000,..
파이썬, 딥러닝 손실함수 교차엔트로피손실(CEE) 개념정리
딥러닝 모델을 구현할 때, 모델을 통해 예상한 값과 실제 값의 LOSS를 정의할때 MSE_LOSS와 CROSS_ENTROYPE_LOSS가 많이 사용된다. 120*120크기의 이미지를 받아 이 이미지가 사과이면 0, 바나나면 1, 메론이면 2 라고 분류하는 모듈을 만들었다고 치자. 그렇다면, y_pred=tensor([[a,b,c]....*배치의 개수])의 형식으로 나올 것이고 a=사과일 확률,b=바나나일 확률, c=메론일 확률 이게 된다. 하지만, y_pred는 엄연한 추정 값이다. 예를들어, a=0.1 b=0.6 c=0.2 라고 하면 우리는 이 그림을 '바나나' 라고 추정을 하게 되는 것이다. 그렇다면 우리는 이 y_pred를 loss=nn.CrossEntroypeLoss(y_pr..