5-1강 CNN코드에서
309
작성한 질문수 8
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(3, 6, 5) # 합성곱 연산 (입력 채널수 3, 출력 채널수 6, 필터크기 5x5 , stride=1(defualt))
self.pool1 = nn.MaxPool2d(2, 2) # 합성곱 연산 (필터크기 2, stride=2)
self.conv2 = nn.Conv2d(6, 16, 5) # 합성곱 연산 (입력 채널수 6, 출력 채널수 16, 필터크기 5x5 , stride=1(defualt))
self.pool2 = nn.MaxPool2d(2, 2) # 합성곱 연산 (필터크기 2, stride=2)
self.fc1 = nn.Linear(16 * 5 * 5, 120) # 5x5 피쳐맵 16개를 일렬로 피면 16*5*5개의 노드가 생성
self.fc2 = nn.Linear(120, 10) # 120개 노드에서 클래스의 개수인 10개의 노드로 연산
def forward(self, x):
x = self.pool1(F.relu(self.conv1(x))) # conv1 -> ReLU -> pool1
x = self.pool2(F.relu(self.conv2(x))) # conv2 -> ReLU -> pool2
x = x.view(-1, 16 * 5 * 5) # 5x5 피쳐맵 16개를 일렬로 만든다.
x = F.relu(self.fc1(x))
x = self.fc2(x)
return x
net = Net().to(device) # 모델 선언에 x = x.view(-1, 16 * 5 * 5) # 5x5 피쳐맵 16개를 일렬로 만든다.
에서 -1이 배치의 개수라고 하셨는데 그럼 이 파일의 코드에서
trainloader = torch.utils.data.DataLoader(trainset, batch_size=8, shuffle=True)trainloader에서 batch_size 가 8이니까 -1값이 제일 마지막에 나누어 떨어지지 않을 때 제외하고
8 인건가요??
아니면
print(len(trainloader))하니까 6250이 나오는데 6250일까요..
배치의 크기를 의미하는건지 헷갈립니다..
답변 1
수업자료 제공 부탁드립니다.
0
95
2
코드가 어디에 있는지 모르겠습니다.
0
95
2
논문 구현
0
188
2
overfitting이 나는 이유가 궁금합니다.
1
171
2
분류 성능이 잘 안 나오는 이유
0
218
1
AutoEncoder 차원 질문
1
250
2
사전 학습 모델에서의 layer 변경에 대한 질문
1
221
1
7강 폴더 만들
0
260
1
4-3강 cross-validation에서의 best model 선정 기준
0
450
1
regression 문제에 대한 결과 시각화
0
213
1
Loss function 관련하여 질문드립니다.
0
957
1
early stopping 코드 문의
0
327
1
예측 그래프
0
355
1
데이터 불균형
1
364
1
8강 전이 학습 질문
0
397
2
data의 gpu처리 질문
0
242
1
nn.Linear(1024, 10) 관련 질문드립니다.
0
288
1
학습과 평가시 Loss 함수가 다른 이유
0
249
1
전처리 관련해서 질문 있습니다.
0
225
1
데이터 엔지니어의 역량을 기르려면 어떻게 해야할까요?
0
935
2
역전파 내용 중 미분 관련 질문 드립니다
1
282
1
8강 전이학습에서 kernel size 관련 질문 드립니다.
1
959
1
이미지분류-합성곱신경망(CNN) 피쳐맵 질문입니다.
1
586
1
14강 데이터 불균형 RandomRotation
1
482
1





