Training the Torchvision's VisionTransformer model using Silicon Mac(arm64, M1/M2)

torch, torchvision
Thonny : opensource editor for python
Mac OSX m1 or m2 machine
VisionTransformer model : https://pytorch.org/vision/stable/models/vision_transformer.html

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
.DS_Store		.DS_Store
README.md		README.md
gpu.png		gpu.png
htop.png		htop.png
thonny_infer.png		thonny_infer.png
torchvision_ViT_finetune_& infer_mps.py		torchvision_ViT_finetune_& infer_mps.py

Provide feedback