#PiT

pit_b_224.in1k是一个基于池化的视觉Transformer(PiT)图像分类模型，在ImageNet-1k数据集上训练。该模型拥有7380万个参数，计算量为12.4 GMACs，支持224x224像素的输入图像。它可用于图像分类、特征图提取和图像嵌入等任务，在保持高精度的同时提高了计算效率。研究人员和开发者可通过timm库轻松加载并使用这个预训练模型。