metadata
tags:
- masked-image-modeling
- generated_from_trainer
datasets:
- cifar10
model-index:
- name: vit-cifar10
results: []
ViT pre-trained from scratch on CIFAR10
This model is a ViT (with the same arch as Google's vit-base-patch16-224 pre-trained from scratch on the cifar10 dataset for masked image modeling.
It achieves the following results on the evaluation set:
- Loss: 0.0891
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 1337
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 100.0
Training results
| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.289 | 1.0 | 2657 | 0.2941 |
| 0.2858 | 2.0 | 5314 | 0.2809 |
| 0.2693 | 3.0 | 7971 | 0.2738 |
| 0.2578 | 4.0 | 10628 | 0.2546 |
| 0.2211 | 5.0 | 13285 | 0.2153 |
| 0.1799 | 6.0 | 15942 | 0.1795 |
| 0.158 | 7.0 | 18599 | 0.1623 |
| 0.1481 | 8.0 | 21256 | 0.1453 |
| 0.1391 | 9.0 | 23913 | 0.1368 |
| 0.1348 | 10.0 | 26570 | 0.1354 |
| 0.129 | 11.0 | 29227 | 0.1249 |
| 0.126 | 12.0 | 31884 | 0.1229 |
| 0.1216 | 13.0 | 34541 | 0.1184 |
| 0.1175 | 14.0 | 37198 | 0.1185 |
| 0.1137 | 15.0 | 39855 | 0.1146 |
| 0.1125 | 16.0 | 42512 | 0.1117 |
| 0.1112 | 17.0 | 45169 | 0.1100 |
| 0.1108 | 18.0 | 47826 | 0.1089 |
| 0.1061 | 19.0 | 50483 | 0.1070 |
| 0.1073 | 20.0 | 53140 | 0.1076 |
| 0.1066 | 21.0 | 55797 | 0.1061 |
| 0.1065 | 22.0 | 58454 | 0.1056 |
| 0.1045 | 23.0 | 61111 | 0.1037 |
| 0.1052 | 24.0 | 63768 | 0.1055 |
| 0.102 | 25.0 | 66425 | 0.1028 |
| 0.1025 | 26.0 | 69082 | 0.1034 |
| 0.1037 | 27.0 | 71739 | 0.1025 |
| 0.1022 | 28.0 | 74396 | 0.1014 |
| 0.1026 | 29.0 | 77053 | 0.1011 |
| 0.1022 | 30.0 | 79710 | 0.1001 |
| 0.0997 | 31.0 | 82367 | 0.1007 |
| 0.0998 | 32.0 | 85024 | 0.1016 |
| 0.1019 | 33.0 | 87681 | 0.1008 |
| 0.0999 | 34.0 | 90338 | 0.1000 |
| 0.0998 | 35.0 | 92995 | 0.0993 |
| 0.0994 | 36.0 | 95652 | 0.0992 |
| 0.0966 | 37.0 | 98309 | 0.0991 |
| 0.0997 | 38.0 | 100966 | 0.0970 |
| 0.0991 | 39.0 | 103623 | 0.0979 |
| 0.099 | 40.0 | 106280 | 0.0983 |
| 0.0974 | 41.0 | 108937 | 0.0980 |
| 0.0974 | 42.0 | 111594 | 0.0971 |
| 0.0972 | 43.0 | 114251 | 0.0970 |
| 0.0991 | 44.0 | 116908 | 0.0970 |
| 0.0979 | 45.0 | 119565 | 0.0972 |
| 0.097 | 46.0 | 122222 | 0.0970 |
| 0.0936 | 47.0 | 124879 | 0.0967 |
| 0.0948 | 48.0 | 127536 | 0.0967 |
| 0.0974 | 49.0 | 130193 | 0.0954 |
| 0.0958 | 50.0 | 132850 | 0.0954 |
| 0.0948 | 51.0 | 135507 | 0.0955 |
| 0.095 | 52.0 | 138164 | 0.0953 |
| 0.0939 | 53.0 | 140821 | 0.0945 |
| 0.0961 | 54.0 | 143478 | 0.0948 |
| 0.0964 | 55.0 | 146135 | 0.0955 |
| 0.0934 | 56.0 | 148792 | 0.0948 |
| 0.0965 | 57.0 | 151449 | 0.0943 |
| 0.0966 | 58.0 | 154106 | 0.0941 |
| 0.0926 | 59.0 | 156763 | 0.0938 |
| 0.0928 | 60.0 | 159420 | 0.0942 |
| 0.093 | 61.0 | 162077 | 0.0936 |
| 0.0939 | 62.0 | 164734 | 0.0939 |
| 0.0936 | 63.0 | 167391 | 0.0936 |
| 0.093 | 64.0 | 170048 | 0.0929 |
| 0.0929 | 65.0 | 172705 | 0.0930 |
| 0.0917 | 66.0 | 175362 | 0.0925 |
| 0.0948 | 67.0 | 178019 | 0.0932 |
| 0.0931 | 68.0 | 180676 | 0.0927 |
| 0.0911 | 69.0 | 183333 | 0.0922 |
| 0.0923 | 70.0 | 185990 | 0.0924 |
| 0.0923 | 71.0 | 188647 | 0.0923 |
| 0.0929 | 72.0 | 191304 | 0.0919 |
| 0.0916 | 73.0 | 193961 | 0.0923 |
| 0.0927 | 74.0 | 196618 | 0.0921 |
| 0.0907 | 75.0 | 199275 | 0.0922 |
| 0.0927 | 76.0 | 201932 | 0.0919 |
| 0.0925 | 77.0 | 204589 | 0.0913 |
| 0.0921 | 78.0 | 207246 | 0.0917 |
| 0.0895 | 79.0 | 209903 | 0.0912 |
| 0.0916 | 80.0 | 212560 | 0.0914 |
| 0.09 | 81.0 | 215217 | 0.0909 |
| 0.0916 | 82.0 | 217874 | 0.0908 |
| 0.0902 | 83.0 | 220531 | 0.0907 |
| 0.0911 | 84.0 | 223188 | 0.0910 |
| 0.091 | 85.0 | 225845 | 0.0903 |
| 0.0903 | 86.0 | 228502 | 0.0905 |
| 0.0907 | 87.0 | 231159 | 0.0901 |
| 0.0908 | 88.0 | 233816 | 0.0907 |
| 0.0911 | 89.0 | 236473 | 0.0902 |
| 0.0905 | 90.0 | 239130 | 0.0906 |
| 0.089 | 91.0 | 241787 | 0.0901 |
| 0.0908 | 92.0 | 244444 | 0.0896 |
| 0.0894 | 93.0 | 247101 | 0.0892 |
| 0.0899 | 94.0 | 249758 | 0.0893 |
| 0.0899 | 95.0 | 252415 | 0.0897 |
| 0.0904 | 96.0 | 255072 | 0.0898 |
| 0.0906 | 97.0 | 257729 | 0.0894 |
| 0.0892 | 98.0 | 260386 | 0.0894 |
| 0.0881 | 99.0 | 263043 | 0.0892 |
| 0.09 | 100.0 | 265700 | 0.0894 |
Framework versions
- Transformers 4.19.0.dev0
- Pytorch 1.10.0+cu111
- Datasets 2.0.0
- Tokenizers 0.11.6