Try ResNet50 instead of VGG. Reason: https://www.researchgate.net/figure/Comparison-of-popular-CNN-architectures-The-vertical-axis-shows-top-1-accuracy-on_fig2_320084139 and https://arxiv.org/pdf/2006.12655.pdf
