model_ft = models.resnet18(weights='IMAGENET1K_V1') num_ftrs = model_ft.fc.in_features # 여기서 각 출력 샘플의 크기는 2로 설정합니다. # 또는, ``nn.Linear(num_ftrs, len (class_names))`` 로 일반화할 수 있습니다. model_ft.fc = nn.Linear(num_ftrs, 2)

여기서

 model_ft.fc1 = nn.Linear(num_ftrs, 1000)

 model_ft.fc2 = nn.Linear(num_ftrs, 2)


이런식으로 완전연결층 게속 못쌓나요?