vit_train_demo.py 6.69 KB