from flowvision.models.swin_transformer import swin_tiny_patch4_window7_224
from flowvision.models.pvt import pvt_tiny
from flowvision.models.cswin import cswin_tiny_224
from flowvision.models.crossformer import crossformer_tiny_patch4_group7_224

model = crossformer_tiny_patch4_group7_224()

import oneflow as flow

x = flow.randn(1, 3, 224 ,224)
output = model(x)
print(output.shape)  # oneflow.Size([1, 1000])