pretrained model layer1 [Pytorch, Huggingface] Pretrained Model 의 특정 Layer 만 추출 목차모델 로드특정 Layer 만을 추출backbone 모델추출구체적인 특정 layer만 추출 1. 모델 로드from transformers import DetrForObjectDetectionmodel_DETR= DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", revision="no_timm")# Model 의 key 값 확인for i in model_DETR.state_dict().keys(): print(i) → model 의 key 값 출력 결과model.backbone.conv_encoder.model.embedder.embedder.convolution.weightmodel.backbone.conv_encoder.m.. 2024. 4. 26. 이전 1 다음