Downloads · 30 days
0
TrevorDohm/ViT_Scratch_MNIST
ViT_Scratch_MNIST is a machine learning model from TrevorDohm. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as openrail.
Trained From Scratch, MNIST https://huggingface.co/datasets/mnist
Downloads · 30 days
0
Access
Public
Updated Mar 7, 2024
Repo size
904 KB
Likes
0
Public
Click a slice to open those files.
.safetensors904 KB · 100%
From the Hugging Face model README
Trained From Scratch, MNIST https://huggingface.co/datasets/mnist
ViT_Small: {"chw": (1, 28, 28), "n_patches": 7, "n_blocks": 4, "hidden_d": 8, "n_heads": 4, "out_d": 10} 23 kB 2K+ Steps
ViT_Large: {"chw": (1, 28, 28), "n_patches": 7, "n_blocks": 6, "hidden_d": 64, "n_heads": 8, "out_d": 10} 881 kB 20K+ Steps
