NYCU Intro. ML final project
- Accuracy 85.9% on both public and private leaderboards
- Leaderboard on Kaggle contest
- Training environment
- Python version: 3.10.13
- framework: Pytorch
- Hardware: NVIDIA 1080Ti with CUDA 12.3 & Intel i7-12700
- Pretrained model
google/vit-base-patch16-224 - Hyper-parameters
- epoch: 5
- batch size: 10
- learning rate: 2e-5
- weight decay: 1e-4