CeiT-pytorch:卷积增强图像变压器的实现

CeiT：卷积增强图像变压器 这是将的非官方PyTorch实现。 训练 ： python train.py -c configs/default.yaml --name "name_of_exp" 用法 ： import torch from ceit import CeiT img = torch . ones ([ 1 , 3 , 224 , 224 ]) model = CeiT ( image_size = 224 , patch_size = 4 , num_classes = 100 ) out = model ( img ) print ( "Shape of out :" , out . shape ) # [B, num_classes] model = CeiT ( image_size = 224 , patch_size = 4 , nu