python-kr.dev
python
pytorch
pandas
django
numpy
sqlalchemy
dataframe
arrays
list
deep-learning
Dark
resnet
[1/1]
PyTorch에서 ResNet 모델의 마지막 FC 레이어를 제거하는 방법
1. 모델 구조를 직접 수정하는 방법torchvision. models에서 원하는 ResNet 모델을 불러옵니다.모델의 children() 메서드를 사용하여 레이어 목록을 가져옵니다.마지막 FC 레이어를 목록에서 제거합니다
1