Model implementations with various configurations (native ViT, ResNet+ViT hybrid, different patch/heads/blocks setups, Stochastic Depth/DropPath, etc.) Training and ...
Train a Convolutional Neural Network (CNN) on the MNIST dataset using PyTorch. 💻 Colab-friendly, supports GPU acceleration, and includes training, evaluation, and visualization. 📂 Project Structure ...