timm documentation

XCiT

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.31).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

XCiT

카테고리: ViT · 코드: timm/models/xcit.py · 아키텍처: 28 · 사전학습된 가중치: 42 (ImageNet-1k 결과 포함 42개)

Cross-Covariance Image Transformers(XCiT)는 토큰 자기 주의를 특징 채널 간 교차 공분산 주의로 대체하며, 이는 토큰 수에 대해 선형 복잡도를 가지며 로컬 패치 상호작용 레이어를 추가합니다.

논문

사용

사전 학습된 가중치 로드, 이미지 전처리 및 파인튜닝에 대해서는 Quickstart를, 모델을 백본으로 사용하는 방법에 대해서는 Feature Extraction을 참고하세요.

>>> import timm
>>> model = timm.create_model('xcit_small_12_p8_224.fb_dist_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

모델 API 지원

기능Xcit
features_only=True feature map✓ (strides 8, 8, 8)
forward_intermediates()✓
prune_intermediate_layers()✓
입력 크기 무관, 변경 불필요✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
레이어별 LR 감쇠(group_matcher())✓
reset_classifier()✓

✓ 지원됨, ✓ (정사각형) 정사각형 크기만 작동, ✗ 미지원, – 필요 없음(모델이 이미 모든 입력 크기를 허용). 무작위로 초기화된 모델(Xcit의 xcit_nano_12_p8_224, 224)에서 각 기능을 실행하여 확인했습니다. 같은 클래스의 다른 아키텍처도 동일한 동작을 하는 것으로 예상됩니다. 자세한 내용은 Model API 페이지를 참조하세요.

아키텍처

아키텍처매개변수 (M)GMAC입력 크기사전 학습된 가중치
xcit_large_24_p8_224188.93141.232242
xcit_large_24_p8_384188.93415.003841
xcit_large_24_p16_224189.1035.862242
xcit_large_24_p16_384189.10105.353841
xcit_medium_24_p8_22484.3263.532242
xcit_medium_24_p8_38484.32186.673841
xcit_medium_24_p16_22484.4016.132242
xcit_medium_24_p16_38484.4047.393841
xcit_nano_12_p8_2243.052.162242
xcit_nano_12_p8_3843.056.343841
xcit_nano_12_p16_2243.050.562242
xcit_nano_12_p16_3843.051.643841
xcit_small_12_p8_22426.2118.692242
xcit_small_12_p8_38426.2154.923841
xcit_small_12_p16_22426.254.822242
xcit_small_12_p16_38426.2514.143841
xcit_small_24_p8_22447.6335.812242
xcit_small_24_p8_38447.63105.243841
xcit_small_24_p16_22447.679.102242
xcit_small_24_p16_38447.6726.723841
xcit_tiny_12_p8_2246.714.812242
xcit_tiny_12_p8_3846.7114.133841
xcit_tiny_12_p16_2246.721.242242
xcit_tiny_12_p16_3846.723.643841
xcit_tiny_24_p8_22412.119.212242
xcit_tiny_24_p8_38412.1127.053841
xcit_tiny_24_p16_22412.122.342242
xcit_tiny_24_p16_38412.126.873841

GMACs와 매개변수 수는 각 아키텍처의 기본 입력 크기 및 분류기에 대한 값이며, 벤치마크 결과에서 가져왔습니다.

사전 학습된 가중치

Top-1은 results-imagenet.csv에 나열된 평가 크기에서 측정된 ImageNet-1k 검증 정확도입니다. - 가중치에 ImageNet-1k 결과가 없는 경우(예: 사전 학습 또는 이미지-텍스트 가중치)에는 표시하지 않습니다. Classes는 사전 학습된 분류기의 크기를 나타내며, 분류기가 없는 가중치의 경우 none입니다.

가중치입력 크기Top-1(%)평가 크기클래스라이선스Hub
xcit_large_24_p8_224.fb_dist_in1k22485.3962241000apache-2.0timm
xcit_large_24_p8_224.fb_in1k22484.3802241000apache-2.0timm
xcit_large_24_p8_384.fb_dist_in1k38486.0243841000apache-2.0timm
xcit_large_24_p16_224.fb_dist_in1k22484.9362241000apache-2.0timm
xcit_large_24_p16_224.fb_in1k22482.8862241000apache-2.0timm
xcit_large_24_p16_384.fb_dist_in1k38485.7783841000apache-2.0timm
xcit_medium_24_p8_224.fb_dist_in1k22485.0782241000apache-2.0timm
xcit_medium_24_p8_224.fb_in1k22483.7442241000apache-2.0timm
xcit_medium_24_p8_384.fb_dist_in1k38485.8423841000apache-2.0timm
xcit_medium_24_p16_224.fb_dist_in1k22484.2922241000apache-2.0timm
xcit_medium_24_p16_224.fb_in1k22482.6302241000apache-2.0timm
xcit_medium_24_p16_384.fb_dist_in1k38485.4343841000apache-2.0timm
xcit_nano_12_p8_224.fb_dist_in1k22476.3302241000apache-2.0timm
xcit_nano_12_p8_224.fb_in1k22473.9002241000apache-2.0timm
xcit_nano_12_p8_384.fb_dist_in1k38477.8103841000apache-2.0timm
xcit_nano_12_p16_224.fb_dist_in1k22472.3182241000apache-2.0timm
xcit_nano_12_p16_224.fb_in1k22469.9822241000apache-2.0timm
xcit_nano_12_p16_384.fb_dist_in1k38475.4603841000apache-2.0timm
xcit_small_12_p8_224.fb_dist_in1k22484.2522241000apache-2.0timm
xcit_small_12_p8_224.fb_in1k22483.3542241000apache-2.0timm
xcit_small_12_p8_384.fb_dist_in1k38485.0683841000apache-2.0timm
xcit_small_12_p16_224.fb_dist_in1k22483.3542241000apache-2.0timm
xcit_small_12_p16_224.fb_in1k22481.9822241000apache-2.0timm
xcit_small_12_p16_384.fb_dist_in1k38484.7343841000apache-2.0timm
xcit_small_24_p8_224.fb_dist_in1k22484.8782241000apache-2.0timm
xcit_small_24_p8_224.fb_in1k22483.8542241000apache-2.0timm
xcit_small_24_p8_384.fb_dist_in1k38485.5803841000apache-2.0timm
xcit_small_24_p16_224.fb_dist_in1k22483.8802241000apache-2.0timm
xcit_small_24_p16_224.fb_in1k22482.5882241000apache-2.0timm
xcit_small_24_p16_384.fb_dist_in1k38485.1023841000apache-2.0timm
xcit_tiny_12_p8_224.fb_dist_in1k22481.2142241000apache-2.0timm
xcit_tiny_12_p8_224.fb_in1k22479.7102241000apache-2.0timm
xcit_tiny_12_p8_384.fb_dist_in1k38482.3943841000apache-2.0timm
xcit_tiny_12_p16_224.fb_dist_in1k22478.5722241000apache-2.0timm
xcit_tiny_12_p16_224.fb_in1k22477.1222241000apache-2.0timm
xcit_tiny_12_p16_384.fb_dist_in1k38480.9543841000apache-2.0timm
xcit_tiny_24_p8_224.fb_dist_in1k22482.5782241000apache-2.0timm
xcit_tiny_24_p8_224.fb_in1k22481.8882241000apache-2.0timm
xcit_tiny_24_p8_384.fb_dist_in1k38483.7843841000apache-2.0timm
xcit_tiny_24_p16_224.fb_dist_in1k22480.4622241000apache-2.0timm
xcit_tiny_24_p16_224.fb_in1k22479.4742241000apache-2.0timm
xcit_tiny_24_p16_384.fb_dist_in1k38482.5803841000apache-2.0timm
Update on GitHub