vit_imagenet.py 1.63 KB