반응형
swin
-
cifar10 Swin Transformer (Base) 학습 - 정리모델 학습 2024. 10. 13. 15:57
모델 학습에 관련된 환경 셋팅 및 코드 정리 오늘은 Swin, vit, ResNet, EfficientNet-B0 중 Swin 에 대해 정리 해보겠습니다.회사에서 업무를 진행하며 정리 하는것이라 내용이 부족할수도 있습니다. pytorch Swin Transformer (Base)https://pytorch.org/vision/main/models/swin_transformer.html SwinTransformer — Torchvision main documentationShortcutspytorch.org Swin Transformer의 주요 특징윈도우 기반 어텐션(Window-based Attention): 전체 이미지에 대해 어텐션을 계산하는 것이 아니라, 작은 윈도우(서브 이미지)를 나누어 지역적인..