Number of data in each class
tablets: 19478
capsules: 4899
ResNet18 pretrained - 69.744
ResNet18 not pretrained
mobilenetv3_large_100_miil - 77.916
ResNet50 - 80.376
efficientnet_b1 - 78.796
tf_efficientnet_b1 - 78.828
vit_small_patch16_224 - 81.396 (bs: 16)
but, 확실히 많이 느려짐. 획기적인 성능 향상이 있지 않는 한 사용하지는 않을듯