Dataset
- UpDown dataset is created using the CIFAR10 dataset
Model Information
- Finetune the 'google/vit-base-patch16-224-in21k' pretrained model for 1 epoch
Performance Measurement
- Binary Cross-Entropy loss is used to measure the training loss
- Accuracy is used to measure the overall model performance in the test set