VGGNet 모델 관련 질문

[ysdo89]

안녕하세요. VggNet 모델 관련해서 질문 드리고 싶은 것이 있어서 글을 작성합니다.

# 훈련 데이터셋과 검증 데이터셋 결합
combined_ds = torch.utils.data.ConcatDataset([train_tmp, val_tmp])

# 결합된 데이터셋을 다시 훈련 데이터셋과 검증 데이터셋으로 분할
# 전체 개수에서 %로 들어갈 수 있도록 조정
train_set_rete = 0.8
train_size = (int)(len(combined_ds) * train_set_rete)

val_size = len(combined_ds) - train_size
print("combined_ds :", len(combined_ds))
train_ds, val_ds = torch.utils.data.random_split(combined_ds, [train_size, val_size])

STL10이 좀 train쪽이 데이터가 적은 것 같아서 데이터를 합친 다음 다시 넣어서 돌려봤는데,

def forward(self, x):
        x = self.conv_layers(x)
        x = x.view(-1, 512 * 7 * 7)
        x = self.fcs(x)
        return x

x = x.view(-1, 512 * 7 * 7) -> 이 부분에서

shape '[-1, 25088]' is invalid for input of size 147456

와 같은 Error 가 발생합니다.

혹시 원인이 뭔지 알 수 있을까요?